ultralytics 8.3.160__tar.gz → 8.3.161__tar.gz

{ultralytics-8.3.160/ultralytics.egg-info → ultralytics-8.3.161}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics
-Version: 8.3.160
+Version: 8.3.161
 Summary: Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics-8.3.160 → ultralytics-8.3.161}/tests/test_python.py

@@ -16,6 +16,7 @@ from tests import CFG, MODEL, MODELS, SOURCE, SOURCES_LIST, TASK_MODEL_DATA, TMP
 from ultralytics import RTDETR, YOLO
 from ultralytics.cfg import TASK2DATA, TASKS
 from ultralytics.data.build import load_inference_source
+from ultralytics.data.utils import check_det_dataset
 from ultralytics.utils import (
     ARM64,
     ASSETS,
@@ -720,7 +721,7 @@ def test_grayscale(task: str, model: str, data: str) -> None:
     if task == "classify":  # not support grayscale classification yet
         return
     grayscale_data = Path(TMP) / f"{Path(data).stem}-grayscale.yaml"
-    data = YAML.load(checks.check_file(data))
+    data = check_det_dataset(data)
     data["channels"] = 1  # add additional channels key for grayscale
     YAML.save(grayscale_data, data)
     # remove npy files in train/val splits if exists, might be created by previous tests

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-__version__ = "8.3.160"
+__version__ = "8.3.161"
 
 import os
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/Argoverse.yaml

@@ -9,7 +9,7 @@
 #     └── Argoverse  ← downloads here (31.5 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/Argoverse # dataset root dir
+path: Argoverse # dataset root dir
 train: Argoverse-1.1/images/train/ # train images (relative to 'path') 39384 images
 val: Argoverse-1.1/images/val/ # val images (relative to 'path') 15062 images
 test: Argoverse-1.1/images/test/ # test images (optional) https://eval.ai/web/challenges/challenge-page/800/overview

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/DOTAv1.5.yaml

@@ -9,7 +9,7 @@
 #     └── dota1.5  ← downloads here (2GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/DOTAv1.5 # dataset root dir
+path: DOTAv1.5 # dataset root dir
 train: images/train # train images (relative to 'path') 1411 images
 val: images/val # val images (relative to 'path') 458 images
 test: images/test # test images (optional) 937 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/DOTAv1.yaml

@@ -9,7 +9,7 @@
 #     └── dota1  ← downloads here (2GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/DOTAv1 # dataset root dir
+path: DOTAv1 # dataset root dir
 train: images/train # train images (relative to 'path') 1411 images
 val: images/val # val images (relative to 'path') 458 images
 test: images/test # test images (optional) 937 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/GlobalWheat2020.yaml

@@ -9,7 +9,7 @@
 #     └── GlobalWheat2020  ← downloads here (7.0 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/GlobalWheat2020 # dataset root dir
+path: GlobalWheat2020 # dataset root dir
 train: # train images (relative to 'path') 3422 images
   - images/arvalis_1
   - images/arvalis_2

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/HomeObjects-3K.yaml

@@ -9,7 +9,7 @@
 #     └── homeobjects-3K  ← downloads here (390 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/homeobjects-3K # dataset root dir
+path: homeobjects-3K # dataset root dir
 train: train/images # train images (relative to 'path') 2285 images
 val: valid/images # val images (relative to 'path') 404 images
 test: # test images (relative to 'path')

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/ImageNet.yaml

@@ -10,7 +10,7 @@
 #     └── imagenet  ← downloads here (144 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/imagenet # dataset root dir
+path: imagenet # dataset root dir
 train: train # train images (relative to 'path') 1281167 images
 val: val # val images (relative to 'path') 50000 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/Objects365.yaml

@@ -9,7 +9,7 @@
 #     └── Objects365  ← downloads here (712 GB = 367G data + 345G zips)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/Objects365 # dataset root dir
+path: Objects365 # dataset root dir
 train: images/train # train images (relative to 'path') 1742289 images
 val: images/val # val images (relative to 'path') 80000 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/SKU-110K.yaml

@@ -9,7 +9,7 @@
 #     └── SKU-110K  ← downloads here (13.6 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/SKU-110K # dataset root dir
+path: SKU-110K # dataset root dir
 train: train.txt # train images (relative to 'path')  8219 images
 val: val.txt # val images (relative to 'path')  588 images
 test: test.txt # test images (optional)  2936 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/VOC.yaml

@@ -9,7 +9,7 @@
 #     └── VOC  ← downloads here (2.8 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/VOC
+path: VOC
 train: # train images (relative to 'path')  16551 images
   - images/train2012
   - images/train2007

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/VisDrone.yaml

@@ -9,7 +9,7 @@
 #     └── VisDrone  ← downloads here (2.3 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/VisDrone # dataset root dir
+path: VisDrone # dataset root dir
 train: VisDrone2019-DET-train/images # train images (relative to 'path')  6471 images
 val: VisDrone2019-DET-val/images # val images (relative to 'path')  548 images
 test: VisDrone2019-DET-test-dev/images # test images (optional)  1610 images
@@ -58,8 +58,11 @@ download: |
                   cls = int(row[5]) - 1
                   box = convert_box(img_size, tuple(map(int, row[:4])))
                   lines.append(f"{cls} {' '.join(f'{x:.6f}' for x in box)}\n")
-                  with open(str(f).replace(f"{os.sep}annotations{os.sep}", f"{os.sep}labels{os.sep}"), "w", encoding="utf-8") as fl:
-                      fl.writelines(lines)  # write label.txt
+          
+          label_file = str(f).replace(f"{os.sep}annotations{os.sep}", f"{os.sep}labels{os.sep}")
+          with open(label_file, "w", encoding="utf-8") as fl:
+                fl.writelines(lines)
+                  
 
 
   # Download

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/african-wildlife.yaml

@@ -9,7 +9,7 @@
 #     └── african-wildlife  ← downloads here (100 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/african-wildlife # dataset root dir
+path: african-wildlife # dataset root dir
 train: train/images # train images (relative to 'path') 1052 images
 val: valid/images # val images (relative to 'path') 225 images
 test: test/images # test images (relative to 'path') 227 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/brain-tumor.yaml

@@ -9,7 +9,7 @@
 #     └── brain-tumor  ← downloads here (4.05 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/brain-tumor # dataset root dir
+path: brain-tumor # dataset root dir
 train: train/images # train images (relative to 'path') 893 images
 val: valid/images # val images (relative to 'path') 223 images
 test: # test images (relative to 'path')

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/carparts-seg.yaml

@@ -9,7 +9,7 @@
 #     └── carparts-seg  ← downloads here (132 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/carparts-seg # dataset root dir
+path: carparts-seg # dataset root dir
 train: train/images # train images (relative to 'path') 3516 images
 val: valid/images # val images (relative to 'path') 276 images
 test: test/images # test images (relative to 'path') 401 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco-pose.yaml

@@ -9,7 +9,7 @@
 #     └── coco-pose  ← downloads here (20.1 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco-pose # dataset root dir
+path: coco-pose # dataset root dir
 train: train2017.txt # train images (relative to 'path') 56599 images
 val: val2017.txt # val images (relative to 'path') 2346 images
 test: test-dev2017.txt # 20288 of 40670 images, submit to https://codalab.lisn.upsaclay.fr/competitions/7403

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco.yaml

@@ -9,7 +9,7 @@
 #     └── coco  ← downloads here (20.1 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco # dataset root dir
+path: coco # dataset root dir
 train: train2017.txt # train images (relative to 'path') 118287 images
 val: val2017.txt # val images (relative to 'path') 5000 images
 test: test-dev2017.txt # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco128-seg.yaml

@@ -9,7 +9,7 @@
 #     └── coco128-seg  ← downloads here (7 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco128-seg # dataset root dir
+path: coco128-seg # dataset root dir
 train: images/train2017 # train images (relative to 'path') 128 images
 val: images/train2017 # val images (relative to 'path') 128 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco128.yaml

@@ -9,7 +9,7 @@
 #     └── coco128  ← downloads here (7 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco128 # dataset root dir
+path: coco128 # dataset root dir
 train: images/train2017 # train images (relative to 'path') 128 images
 val: images/train2017 # val images (relative to 'path') 128 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco8-grayscale.yaml

@@ -9,7 +9,7 @@
 #     └── coco8-grayscale  ← downloads here (1 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco8-grayscale # dataset root dir
+path: coco8-grayscale # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco8-multispectral.yaml

@@ -9,7 +9,7 @@
 #     └── coco8-multispectral  ← downloads here (20.2 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco8-multispectral # dataset root dir
+path: coco8-multispectral # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco8-pose.yaml

@@ -9,7 +9,7 @@
 #     └── coco8-pose  ← downloads here (1 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco8-pose # dataset root dir
+path: coco8-pose # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco8-seg.yaml

@@ -9,7 +9,7 @@
 #     └── coco8-seg  ← downloads here (1 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco8-seg # dataset root dir
+path: coco8-seg # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/coco8.yaml

@@ -9,7 +9,7 @@
 #     └── coco8  ← downloads here (1 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/coco8 # dataset root dir
+path: coco8 # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/crack-seg.yaml

@@ -9,7 +9,7 @@
 #     └── crack-seg  ← downloads here (91.2 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/crack-seg # dataset root dir
+path: crack-seg # dataset root dir
 train: train/images # train images (relative to 'path') 3717 images
 val: valid/images # val images (relative to 'path') 112 images
 test: test/images # test images (relative to 'path') 200 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/dog-pose.yaml

@@ -9,7 +9,7 @@
 #     └── dog-pose  ← downloads here (337 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/dog-pose # dataset root dir
+path: dog-pose # dataset root dir
 train: train # train images (relative to 'path') 6773 images
 val: val # val images (relative to 'path') 1703 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/dota8-multispectral.yaml

@@ -9,7 +9,7 @@
 #     └── dota8-multispectral  ← downloads here (37.3MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/dota8-multispectral # dataset root dir
+path: dota8-multispectral # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/dota8.yaml

@@ -9,7 +9,7 @@
 #     └── dota8  ← downloads here (1MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/dota8 # dataset root dir
+path: dota8 # dataset root dir
 train: images/train # train images (relative to 'path') 4 images
 val: images/val # val images (relative to 'path') 4 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/hand-keypoints.yaml

@@ -9,7 +9,7 @@
 #     └── hand-keypoints  ← downloads here (369 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/hand-keypoints # dataset root dir
+path: hand-keypoints # dataset root dir
 train: train # train images (relative to 'path') 18776 images
 val: val # val images (relative to 'path') 7992 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/lvis.yaml

@@ -9,7 +9,7 @@
 #     └── lvis  ← downloads here (20.1 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/lvis # dataset root dir
+path: lvis # dataset root dir
 train: train.txt # train images (relative to 'path') 100170 images
 val: val.txt # val images (relative to 'path') 19809 images
 minival: minival.txt # minival images (relative to 'path') 5000 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/medical-pills.yaml

@@ -9,7 +9,7 @@
 #     └── medical-pills  ← downloads here (8.19 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/medical-pills # dataset root dir
+path: medical-pills # dataset root dir
 train: train/images # train images (relative to 'path') 92 images
 val: valid/images # val images (relative to 'path') 23 images
 test: # test images (relative to 'path')

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/open-images-v7.yaml

@@ -9,7 +9,7 @@
 #     └── open-images-v7  ← downloads here (561 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/open-images-v7 # dataset root dir
+path: open-images-v7 # dataset root dir
 train: images/train # train images (relative to 'path') 1743042 images
 val: images/val # val images (relative to 'path') 41620 images
 test: # test images (optional)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/package-seg.yaml

@@ -9,7 +9,7 @@
 #     └── package-seg  ← downloads here (102 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/package-seg # dataset root dir
+path: package-seg # dataset root dir
 train: train/images # train images (relative to 'path') 1920 images
 val: valid/images # val images (relative to 'path') 89 images
 test: test/images # test images (relative to 'path') 188 images

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/signature.yaml

@@ -9,7 +9,7 @@
 #     └── signature  ← downloads here (11.2 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/signature # dataset root dir
+path: signature # dataset root dir
 train: train/images # train images (relative to 'path') 143 images
 val: valid/images # val images (relative to 'path') 35 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/tiger-pose.yaml

@@ -9,7 +9,7 @@
 #     └── tiger-pose  ← downloads here (75.3 MB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/tiger-pose # dataset root dir
+path: tiger-pose # dataset root dir
 train: train # train images (relative to 'path') 210 images
 val: val # val images (relative to 'path') 53 images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/cfg/datasets/xView.yaml

@@ -10,7 +10,7 @@
 #     └── xView  ← downloads here (20.7 GB)
 
 # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
-path: ../datasets/xView # dataset root dir
+path: xView # dataset root dir
 train: images/autosplit_train.txt # train images (relative to 'path') 90% of 847 train images
 val: images/autosplit_val.txt # train images (relative to 'path') 10% of 847 train images
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/data/converter.py

@@ -248,12 +248,10 @@ def convert_coco(
         >>> from ultralytics.data.converter import convert_coco
 
         Convert COCO annotations to YOLO format
-        >>> convert_coco("../datasets/coco/annotations/", use_segments=True, use_keypoints=False, cls91to80=False)
+        >>> convert_coco("coco/annotations/", use_segments=True, use_keypoints=False, cls91to80=False)
 
         Convert LVIS annotations to YOLO format
-        >>> convert_coco(
-        ...     "../datasets/lvis/annotations/", use_segments=True, use_keypoints=False, cls91to80=False, lvis=True
-        ... )
+        >>> convert_coco("lvis/annotations/", use_segments=True, use_keypoints=False, cls91to80=False, lvis=True)
     """
     # Create dataset directory
     save_dir = increment_path(save_dir)  # increment if save directory already exists
@@ -724,7 +722,7 @@ def convert_to_multispectral(path: Union[str, Path], n_channels: int = 10, repla
         >>> convert_to_multispectral("path/to/image.jpg", n_channels=10)
 
         Convert a dataset
-        >>> convert_to_multispectral("../datasets/coco8", n_channels=10)
+        >>> convert_to_multispectral("coco8", n_channels=10)
     """
     from scipy.interpolate import interp1d
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/data/dataset.py

@@ -482,7 +482,7 @@ class GroundingDataset(YOLODataset):
             a warning is logged and verification is skipped.
         """
         expected_counts = {
-            "final_mixed_train_no_coco_segm": 3662344,
+            "final_mixed_train_no_coco_segm": 3662412,
             "final_mixed_train_no_coco": 3681235,
             "final_flickr_separateGT_train_segm": 638214,
             "final_flickr_separateGT_train": 640704,

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/data/split.py

@@ -135,4 +135,4 @@ def autosplit(
 
 
 if __name__ == "__main__":
-    split_classify_dataset("../datasets/caltech101")
+    split_classify_dataset("caltech101")

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/engine/exporter.py

@@ -706,7 +706,16 @@ class Exporter:
     def export_paddle(self, prefix=colorstr("PaddlePaddle:")):
         """Export YOLO model to PaddlePaddle format."""
         assert not IS_JETSON, "Jetson Paddle exports not supported yet"
-        check_requirements(("paddlepaddle-gpu" if torch.cuda.is_available() else "paddlepaddle>=3.0.0", "x2paddle"))
+        check_requirements(
+            (
+                "paddlepaddle-gpu"
+                if torch.cuda.is_available()
+                else "paddlepaddle==3.0.0"  # pin 3.0.0 for ARM64
+                if ARM64
+                else "paddlepaddle>=3.0.0",
+                "x2paddle",
+            )
+        )
         import x2paddle  # noqa
         from x2paddle.convert import pytorch2paddle  # noqa
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/engine/results.py

@@ -800,7 +800,7 @@ class Results(SimpleClass, DataExportMixin):
             decimals (int): Number of decimal places to round the output values to.
 
         Returns:
-            (List[Dict]): A list of dictionaries, each containing summarized information for a single detection
+            (List[Dict[str, Any]]): A list of dictionaries, each containing summarized information for a single detection
                 or classification result. The structure of each dictionary varies based on the task type
                 (classification or detection) and available information (boxes, masks, keypoints).
 

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/models/yolo/world/train_world.py

@@ -35,12 +35,12 @@ class WorldTrainerFromScratch(WorldTrainer):
         ...         yolo_data=["Objects365.yaml"],
         ...         grounding_data=[
         ...             dict(
-        ...                 img_path="../datasets/flickr30k/images",
-        ...                 json_file="../datasets/flickr30k/final_flickr_separateGT_train.json",
+        ...                 img_path="flickr30k/images",
+        ...                 json_file="flickr30k/final_flickr_separateGT_train.json",
         ...             ),
         ...             dict(
-        ...                 img_path="../datasets/GQA/images",
-        ...                 json_file="../datasets/GQA/final_mixed_train_no_coco.json",
+        ...                 img_path="GQA/images",
+        ...                 json_file="GQA/final_mixed_train_no_coco.json",
         ...             ),
         ...         ],
         ...     ),
@@ -70,8 +70,8 @@ class WorldTrainerFromScratch(WorldTrainer):
             ...         yolo_data=["Objects365.yaml"],
             ...         grounding_data=[
             ...             dict(
-            ...                 img_path="../datasets/flickr30k/images",
-            ...                 json_file="../datasets/flickr30k/final_flickr_separateGT_train.json",
+            ...                 img_path="flickr30k/images",
+            ...                 json_file="flickr30k/final_flickr_separateGT_train.json",
             ...             ),
             ...         ],
             ...     ),

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/nn/autobackend.py

@@ -487,7 +487,13 @@ class AutoBackend(nn.Module):
         # PaddlePaddle
         elif paddle:
             LOGGER.info(f"Loading {w} for PaddlePaddle inference...")
-            check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle>=3.0.0")
+            check_requirements(
+                "paddlepaddle-gpu"
+                if torch.cuda.is_available()
+                else "paddlepaddle==3.0.0"  # pin 3.0.0 for ARM64
+                if ARM64
+                else "paddlepaddle>=3.0.0"
+            )
             import paddle.inference as pdi  # noqa
 
             w = Path(w)

{ultralytics-8.3.160 → ultralytics-8.3.161}/ultralytics/solutions/similarity_search.py

@@ -9,14 +9,14 @@ from PIL import Image
 
 from ultralytics.data.utils import IMG_FORMATS
 from ultralytics.nn.text_model import build_text_model
-from ultralytics.solutions.solutions import BaseSolution
+from ultralytics.utils import LOGGER
 from ultralytics.utils.checks import check_requirements
 from ultralytics.utils.torch_utils import select_device
 
 os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"  # Avoid OpenMP conflict on some systems
 
 
-class VisualAISearch(BaseSolution):
+class VisualAISearch:
     """
     A semantic image search system that leverages OpenCLIP for generating high-quality image and text embeddings and
     FAISS for fast similarity-based retrieval.
@@ -48,19 +48,18 @@ class VisualAISearch(BaseSolution):
 
     def __init__(self, **kwargs: Any) -> None:
         """Initialize the VisualAISearch class with FAISS index and CLIP model."""
-        super().__init__(**kwargs)
         check_requirements("faiss-cpu")
 
         self.faiss = __import__("faiss")
         self.faiss_index = "faiss.index"
         self.data_path_npy = "paths.npy"
-        self.data_dir = Path(self.CFG["data"])
-        self.device = select_device(self.CFG["device"])
+        self.data_dir = Path(kwargs.get("data", "images"))
+        self.device = select_device(kwargs.get("device", "cpu"))
 
         if not self.data_dir.exists():
             from ultralytics.utils import ASSETS_URL
 
-            self.LOGGER.warning(f"{self.data_dir} not found. Downloading images.zip from {ASSETS_URL}/images.zip")
+            LOGGER.warning(f"{self.data_dir} not found. Downloading images.zip from {ASSETS_URL}/images.zip")
             from ultralytics.utils.downloads import safe_download
 
             safe_download(url=f"{ASSETS_URL}/images.zip", unzip=True, retry=3)
@@ -91,13 +90,13 @@ class VisualAISearch(BaseSolution):
         """
         # Check if the FAISS index and corresponding image paths already exist
         if Path(self.faiss_index).exists() and Path(self.data_path_npy).exists():
-            self.LOGGER.info("Loading existing FAISS index...")
+            LOGGER.info("Loading existing FAISS index...")
             self.index = self.faiss.read_index(self.faiss_index)  # Load the FAISS index from disk
             self.image_paths = np.load(self.data_path_npy)  # Load the saved image path list
             return  # Exit the function as the index is successfully loaded
 
         # If the index doesn't exist, start building it from scratch
-        self.LOGGER.info("Building FAISS index from images...")
+        LOGGER.info("Building FAISS index from images...")
         vectors = []  # List to store feature vectors of images
 
         # Iterate over all image files in the data directory
@@ -110,7 +109,7 @@ class VisualAISearch(BaseSolution):
                 vectors.append(self.extract_image_feature(file))
                 self.image_paths.append(file.name)  # Store the corresponding image name
             except Exception as e:
-                self.LOGGER.warning(f"Skipping {file.name}: {e}")
+                LOGGER.warning(f"Skipping {file.name}: {e}")
 
         # If no vectors were successfully created, raise an error
         if not vectors:
@@ -124,7 +123,7 @@ class VisualAISearch(BaseSolution):
         self.faiss.write_index(self.index, self.faiss_index)  # Save the newly built FAISS index to disk
         np.save(self.data_path_npy, np.array(self.image_paths))  # Save the list of image paths to disk
 
-        self.LOGGER.info(f"Indexed {len(self.image_paths)} images.")
+        LOGGER.info(f"Indexed {len(self.image_paths)} images.")
 
     def search(self, query: str, k: int = 30, similarity_thresh: float = 0.1) -> List[str]:
         """
@@ -152,9 +151,9 @@ class VisualAISearch(BaseSolution):
         ]
         results.sort(key=lambda x: x[1], reverse=True)
 
-        self.LOGGER.info("\nRanked Results:")
+        LOGGER.info("\nRanked Results:")
         for name, score in results:
-            self.LOGGER.info(f"  - {name} | Similarity: {score:.4f}")
+            LOGGER.info(f"  - {name} | Similarity: {score:.4f}")
 
         return [r[0] for r in results]