ultralytics 8.2.86 Windows torch==2.4.0 patch (#15942)

Signed-off-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-09-01 21:37:05 +02:00 · 2024-09-01 21:37:05 +02:00 · 62408aeab8
commit 62408aeab8
parent 0f105f4ea2
6 changed files with 23 additions and 2 deletions
--- a/tests/test_python.py
+++ b/tests/test_python.py
@ -252,6 +252,8 @@ def test_labels_and_crops():
    for r in results:
        im_name = Path(r.path).stem
        cls_idxs = r.boxes.cls.int().tolist()
+        # Check correct detections
+        assert cls_idxs == ([0, 0, 5, 0, 7] if r.path.endswith("bus.jpg") else [0, 0])  # bus.jpg and zidane.jpg classes
        # Check label path
        labels = save_path / f"labels/{im_name}.txt"
        assert labels.exists()