Unverified 提交 8ca18261 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update `pycocotools>=2.0.6` (#10605)

* Update `pycocotools>=2.0.6` Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update val.py Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update val.py Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update Dockerfile Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 b1e99764
...@@ -44,6 +44,6 @@ seaborn>=0.11.0 ...@@ -44,6 +44,6 @@ seaborn>=0.11.0
# Extras ---------------------------------------------------------------------- # Extras ----------------------------------------------------------------------
# mss # screenshots # mss # screenshots
# albumentations>=1.0.3 # albumentations>=1.0.3
# pycocotools>=2.0 # COCO mAP # pycocotools>=2.0.6 # COCO mAP
# roboflow # roboflow
# ultralytics # HUB https://hub.ultralytics.com # ultralytics # HUB https://hub.ultralytics.com
...@@ -159,7 +159,7 @@ def run( ...@@ -159,7 +159,7 @@ def run(
callbacks=Callbacks(), callbacks=Callbacks(),
): ):
if save_json: if save_json:
check_requirements(['pycocotools']) check_requirements('pycocotools>=2.0.6')
process = process_mask_native # more accurate process = process_mask_native # more accurate
else: else:
process = process_mask # faster process = process_mask # faster
......
...@@ -16,8 +16,8 @@ RUN apt update && apt install --no-install-recommends -y zip htop screen libgl1- ...@@ -16,8 +16,8 @@ RUN apt update && apt install --no-install-recommends -y zip htop screen libgl1-
COPY requirements.txt . COPY requirements.txt .
RUN python -m pip install --upgrade pip wheel RUN python -m pip install --upgrade pip wheel
RUN pip uninstall -y Pillow torchtext torch torchvision RUN pip uninstall -y Pillow torchtext torch torchvision
RUN pip install --no-cache -r requirements.txt ultralytics albumentations comet gsutil notebook Pillow>=9.1.0 \ RUN pip install --no-cache -r requirements.txt albumentations comet gsutil notebook 'opencv-python<4.6.0.66' \
'opencv-python<4.6.0.66' \ Pillow>=9.1.0 pycocotools>=2.0.6 ultralytics \
--extra-index-url https://download.pytorch.org/whl/cu113 --extra-index-url https://download.pytorch.org/whl/cu113
# Create working directory # Create working directory
......
...@@ -309,7 +309,7 @@ def run( ...@@ -309,7 +309,7 @@ def run(
json.dump(jdict, f) json.dump(jdict, f)
try: # https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocoEvalDemo.ipynb try: # https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocoEvalDemo.ipynb
check_requirements('pycocotools') check_requirements('pycocotools>=2.0.6')
from pycocotools.coco import COCO from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval from pycocotools.cocoeval import COCOeval
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论