Pin `albumentations<=1.4.4` to fix missing `bboxes` key bug (#11236)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/11229/head^2
Glenn Jocher 7 months ago committed by GitHub
parent 553fa0e967
commit ccfc1cf925
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      docker/Dockerfile
  2. 2
      pyproject.toml

@ -30,7 +30,7 @@ ADD https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.pt /u
# Install pip packages
RUN python3 -m pip install --upgrade pip wheel
RUN pip install --no-cache -e ".[export]" albumentations comet pycocotools
RUN pip install --no-cache -e ".[export]" "albumentations<=1.4.4" comet pycocotools
# Run exports to AutoInstall packages
# Edge TPU export fails the first time so is run twice here

@ -123,7 +123,7 @@ logging = [
extra = [
"hub-sdk>=0.0.5", # Ultralytics HUB
"ipython", # interactive notebook
"albumentations>=1.0.3", # training augmentations
"albumentations>=1.0.3,<=1.4.4", # training augmentations
"pycocotools>=2.0.7", # COCO mAP
]

Loading…
Cancel
Save