diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 15366b7d..ddd289bc 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -254,14 +254,27 @@ jobs: - name: Install requirements run: | python -m pip install --upgrade pip wheel - pip install -e . pytest mlflow pycocotools "ray[tune]<=2.9.3" --extra-index-url https://download.pytorch.org/whl/cpu + pip install -e ".[export]" pytest mlflow pycocotools "ray[tune]<=2.9.3" - name: Check environment run: | yolo checks pip list - name: Pytest tests + run: pytest --slow tests/ + - name: Benchmark ClassificationModel + run: python -m ultralytics.cfg.__init__ benchmark model='yolov8n-cls.pt' imgsz=160 verbose=0.166 + - name: Benchmark YOLOWorld DetectionModel + run: python -m ultralytics.cfg.__init__ benchmark model='yolov8s-worldv2.pt' imgsz=160 verbose=0.318 + - name: Benchmark SegmentationModel + run: python -m ultralytics.cfg.__init__ benchmark model='yolov8n-seg.pt' imgsz=160 verbose=0.267 + - name: Benchmark PoseModel + run: python -m ultralytics.cfg.__init__ benchmark model='yolov8n-pose.pt' imgsz=160 verbose=0.179 + - name: Benchmark OBBModel + run: python -m ultralytics.cfg.__init__ benchmark model='yolov8n-obb.pt' imgsz=160 verbose=0.472 + - name: Benchmark Summary run: | - pytest --slow tests/ + cat benchmarks.log + echo "$(cat benchmarks.log)" >> $GITHUB_STEP_SUMMARY - name: Reboot # run a reboot command in the background to free resources for next run and not crash main thread run: sudo bash -c "sleep 10; reboot" & diff --git a/pyproject.toml b/pyproject.toml index ad860173..d4368f70 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -102,7 +102,7 @@ export = [ "openvino>=2024.0.0", # OpenVINO export "tensorflow<=2.13.1; python_version <= '3.11'", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 "tensorflowjs>=3.9.0; python_version <= '3.11'", # TF.js export, automatically installs tensorflow - # "flatbuffers>=23.5.26,<100", # update old 'flatbuffers' included inside tensorflow package: WARNING Dockerfile error https://github.com/ultralytics/ultralytics/actions/runs/8715942435/job/23908971614 + "flatbuffers>=23.5.26,<100; platform_machine == 'aarch64'", # update old 'flatbuffers' included inside tensorflow package "numpy==1.23.5; platform_machine == 'aarch64'", # fix error: `np.bool` was a deprecated alias for the builtin `bool` when using TensorRT models on NVIDIA Jetson "h5py!=3.11.0; platform_machine == 'aarch64'", # fix h5py build issues due to missing aarch64 wheels in 3.11 release ] diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 78a48007..bb2ea3f6 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.6" +__version__ = "8.2.7" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index b1a52b0f..d2536542 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -414,7 +414,7 @@ class Exporter: @try_export def export_openvino(self, prefix=colorstr("OpenVINO:")): """YOLOv8 OpenVINO export.""" - check_requirements("openvino>=2024.0.0") # requires openvino: https://pypi.org/project/openvino/ + check_requirements(f'openvino{"<=2024.0.0" if ARM64 else ">=2024.0.0"}') # fix OpenVINO issue on ARM64 import openvino as ov LOGGER.info(f"\n{prefix} starting export with openvino {ov.__version__}...")