diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index dd9e9a351..816a0a321 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -64,7 +64,7 @@ jobs: pip install -e .[dev] # stop the build if there are Python syntax errors or undefined names python -m scripts.run_code_style check - + # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide flake8 . --exit-zero --max-complexity=10 --max-line-length=127 @@ -107,7 +107,7 @@ jobs: - name: Install ultralytics run: > - pip install ultralytics==8.0.43 + pip install ultralytics==8.0.99 - name: Unittest for SAHI+YOLOV5/MMDET/Detectron2 on all platforms run: | diff --git a/.github/workflows/ci_torch1.10.yml b/.github/workflows/ci_torch1.10.yml index f938b4bd1..a130e3e19 100644 --- a/.github/workflows/ci_torch1.10.yml +++ b/.github/workflows/ci_torch1.10.yml @@ -65,7 +65,7 @@ jobs: pip install -e .[dev] # stop the build if there are Python syntax errors or undefined names python -m scripts.run_code_style check - + # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide flake8 . --exit-zero --max-complexity=10 --max-line-length=127 @@ -114,7 +114,7 @@ jobs: - name: Install ultralytics run: > - pip install ultralytics==8.0.43 + pip install ultralytics==8.0.99 - name: Unittest for SAHI+YOLOV5/MMDET/Detectron2 on all platforms run: | diff --git a/.github/workflows/package_testing.yml b/.github/workflows/package_testing.yml index 4bb0f7149..f9b21e023 100644 --- a/.github/workflows/package_testing.yml +++ b/.github/workflows/package_testing.yml @@ -70,11 +70,11 @@ jobs: - name: Install YOLOv5(7.0.9) run: > pip install yolov5==7.0.9 - + - name: Install DeepSparse run: > pip install deepsparse - + - name: Install Detectron2(0.6) run: > python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.10/index.html @@ -89,7 +89,7 @@ jobs: - name: Install ultralytics run: > - pip install ultralytics==8.0.43 + pip install ultralytics==8.0.99 - name: Install latest SAHI package run: > diff --git a/sahi/models/yolov8.py b/sahi/models/yolov8.py index 62c0f1ffb..f8ba492be 100644 --- a/sahi/models/yolov8.py +++ b/sahi/models/yolov8.py @@ -60,7 +60,7 @@ def perform_inference(self, image: np.ndarray): raise ValueError("Model is not loaded, load it by calling .load_model()") prediction_result = self.model(image[:, :, ::-1], verbose=False) # YOLOv8 expects numpy arrays to have BGR prediction_result = [ - result.boxes.boxes[result.boxes.boxes[:, 4] >= self.confidence_threshold] for result in prediction_result + result.boxes.data[result.boxes.data[:, 4] >= self.confidence_threshold] for result in prediction_result ] self._original_predictions = prediction_result