dgenerate-ultralytics-headless 8.3.165__tar.gz → 8.3.166__tar.gz

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.165
+Version: 8.3.166
 Summary: Automatically built Ultralytics package with python-opencv-headless dependency instead of python-opencv
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/dgenerate_ultralytics_headless.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.165
+Version: 8.3.166
 Summary: Automatically built Ultralytics package with python-opencv-headless dependency instead of python-opencv
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/__init__.py

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

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/Argoverse.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── Argoverse  ← downloads here (31.5 GB)
+#     └── 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: Argoverse # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/DOTAv1.5.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── dota1.5  ← downloads here (2GB)
+#     └── 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: DOTAv1.5 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/DOTAv1.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── dota1  ← downloads here (2GB)
+#     └── 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: DOTAv1 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/GlobalWheat2020.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── GlobalWheat2020  ← downloads here (7.0 GB)
+#     └── 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: GlobalWheat2020 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/HomeObjects-3K.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── homeobjects-3K  ← downloads here (390 MB)
+#     └── 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: homeobjects-3K # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/ImageNet.yaml

@@ -7,7 +7,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── imagenet  ← downloads here (144 GB)
+#     └── 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: imagenet # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/Objects365.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── Objects365  ← downloads here (712 GB = 367G data + 345G zips)
+#     └── 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: Objects365 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/SKU-110K.yaml

@@ -6,13 +6,13 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── SKU-110K  ← downloads here (13.6 GB)
+#     └── 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: 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
+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
 
 # Classes
 names:

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/VOC.yaml

@@ -6,16 +6,16 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── VOC  ← downloads here (2.8 GB)
+#     └── 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: VOC
-train: # train images (relative to 'path')  16551 images
+train: # train images (relative to 'path') 16551 images
   - images/train2012
   - images/train2007
   - images/val2012
   - images/val2007
-val: # val images (relative to 'path')  4952 images
+val: # val images (relative to 'path') 4952 images
   - images/test2007
 test: # test images (optional)
   - images/test2007

dgenerate_ultralytics_headless-8.3.166/ultralytics/cfg/datasets/VisDrone.yaml

@@ -0,0 +1,87 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+# VisDrone2019-DET dataset https://github.com/VisDrone/VisDrone-Dataset by Tianjin University
+# Documentation: https://docs.ultralytics.com/datasets/detect/visdrone/
+# Example usage: yolo train data=VisDrone.yaml
+# parent
+# ├── ultralytics
+# └── datasets
+#     └── 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: VisDrone # dataset root dir
+train: images/train # train images (relative to 'path') 6471 images
+val: images/val # val images (relative to 'path') 548 images
+test: images/test # test-dev images (optional) 1610 images
+
+# Classes
+names:
+  0: pedestrian
+  1: people
+  2: bicycle
+  3: car
+  4: van
+  5: truck
+  6: tricycle
+  7: awning-tricycle
+  8: bus
+  9: motor
+
+# Download script/URL (optional) ---------------------------------------------------------------------------------------
+download: |
+  import os
+  from pathlib import Path
+  import shutil
+
+  from ultralytics.utils.downloads import download
+
+
+  def visdrone2yolo(dir, split, source_name=None):
+      """Convert VisDrone annotations to YOLO format with images/{split} and labels/{split} structure."""
+      from PIL import Image
+      from tqdm import tqdm
+
+      source_dir = dir / (source_name or f"VisDrone2019-DET-{split}")
+      images_dir = dir / "images" / split
+      labels_dir = dir / "labels" / split
+      labels_dir.mkdir(parents=True, exist_ok=True)
+      
+      # Move images to new structure
+      if (source_images_dir := source_dir / "images").exists():
+          images_dir.mkdir(parents=True, exist_ok=True)
+          for img in source_images_dir.glob("*.jpg"):
+              img.rename(images_dir / img.name)
+      
+      for f in tqdm((source_dir / "annotations").glob("*.txt"), desc=f"Converting {split}"):
+          img_size = Image.open(images_dir / f.with_suffix(".jpg").name).size
+          dw, dh = 1.0 / img_size[0], 1.0 / img_size[1]
+          lines = []
+          
+          with open(f, encoding="utf-8") as file:
+              for row in [x.split(",") for x in file.read().strip().splitlines()]:
+                  if row[4] != "0":  # Skip ignored regions
+                      x, y, w, h = map(int, row[:4])
+                      cls = int(row[5]) - 1
+                      # Convert to YOLO format
+                      x_center, y_center = (x + w / 2) * dw, (y + h / 2) * dh
+                      w_norm, h_norm = w * dw, h * dh
+                      lines.append(f"{cls} {x_center:.6f} {y_center:.6f} {w_norm:.6f} {h_norm:.6f}\n")
+          
+          (labels_dir / f.name).write_text("".join(lines), encoding="utf-8")
+
+
+  # Download (ignores test-challenge split)
+  dir = Path(yaml["path"])  # dataset root dir
+  urls = [
+      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-train.zip",
+      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-val.zip",
+      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-test-dev.zip",
+      # "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-test-challenge.zip",
+  ]
+  download(urls, dir=dir, curl=True, threads=4)
+
+  # Convert
+  splits = {"VisDrone2019-DET-train": "train", "VisDrone2019-DET-val": "val", "VisDrone2019-DET-test-dev": "test"}
+  for folder, split in splits.items():
+      visdrone2yolo(dir, split, folder)  # convert VisDrone annotations to YOLO labels
+      shutil.rmtree(dir / folder)  # cleanup original directory

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/african-wildlife.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── african-wildlife  ← downloads here (100 MB)
+#     └── 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: african-wildlife # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/brain-tumor.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── brain-tumor  ← downloads here (4.21 MB)
+#     └── brain-tumor ← downloads here (4.21 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: brain-tumor # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/carparts-seg.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── carparts-seg  ← downloads here (133 MB)
+#     └── carparts-seg ← downloads here (133 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: carparts-seg # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco-pose.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco-pose  ← downloads here (20.1 GB)
+#     └── 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: coco-pose # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco  ← downloads here (20.1 GB)
+#     └── 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: coco # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco128-seg.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco128-seg  ← downloads here (7 MB)
+#     └── 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: coco128-seg # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco128.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco128  ← downloads here (7 MB)
+#     └── 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: coco128 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco8-grayscale.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco8-grayscale  ← downloads here (1 MB)
+#     └── 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: coco8-grayscale # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco8-multispectral.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco8-multispectral  ← downloads here (20.2 MB)
+#     └── 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: coco8-multispectral # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco8-pose.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco8-pose  ← downloads here (1 MB)
+#     └── 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: coco8-pose # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco8-seg.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco8-seg  ← downloads here (1 MB)
+#     └── 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: coco8-seg # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/coco8.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── coco8  ← downloads here (1 MB)
+#     └── 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: coco8 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/crack-seg.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── crack-seg  ← downloads here (91.6 MB)
+#     └── crack-seg ← downloads here (91.6 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: crack-seg # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/dog-pose.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── dog-pose  ← downloads here (337 MB)
+#     └── 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: dog-pose # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/dota8-multispectral.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── dota8-multispectral  ← downloads here (37.3MB)
+#     └── 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: dota8-multispectral # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/dota8.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── dota8  ← downloads here (1MB)
+#     └── 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: dota8 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/hand-keypoints.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── hand-keypoints  ← downloads here (369 MB)
+#     └── 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: hand-keypoints # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/lvis.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── lvis  ← downloads here (20.1 GB)
+#     └── 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: lvis # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/medical-pills.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── medical-pills  ← downloads here (8.19 MB)
+#     └── 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: medical-pills # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/open-images-v7.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── open-images-v7  ← downloads here (561 GB)
+#     └── 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: open-images-v7 # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/package-seg.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── package-seg  ← downloads here (103 MB)
+#     └── package-seg ← downloads here (103 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: package-seg # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/signature.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── signature  ← downloads here (11.3 MB)
+#     └── signature ← downloads here (11.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: signature # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/tiger-pose.yaml

@@ -6,7 +6,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── tiger-pose  ← downloads here (22.8 MB)
+#     └── tiger-pose ← downloads here (49.8 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: tiger-pose # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/cfg/datasets/xView.yaml

@@ -7,7 +7,7 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── xView  ← downloads here (20.7 GB)
+#     └── 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: xView # dataset root dir

{dgenerate_ultralytics_headless-8.3.165 → dgenerate_ultralytics_headless-8.3.166}/ultralytics/utils/metrics.py

@@ -330,7 +330,7 @@ class ConfusionMatrix(DataExportMixin):
         """
         self.task = task
         self.nc = len(names)  # number of classes
-        self.matrix = np.zeros((self.nc + 1, self.nc + 1)) if self.task == "detect" else np.zeros((self.nc, self.nc))
+        self.matrix = np.zeros((self.nc, self.nc)) if self.task == "classify" else np.zeros((self.nc + 1, self.nc + 1))
         self.names = names  # name of classes
 
     def process_cls_preds(self, preds, targets):
@@ -360,8 +360,9 @@ class ConfusionMatrix(DataExportMixin):
             conf (float, optional): Confidence threshold for detections.
             iou_thres (float, optional): IoU threshold for matching detections to ground truth.
         """
-        conf = 0.25 if conf in {None, 0.001} else conf  # apply 0.25 if default val conf is passed
         gt_cls, gt_bboxes = batch["cls"], batch["bboxes"]
+        is_obb = gt_bboxes.shape[1] == 5  # check if boxes contains angle for OBB
+        conf = 0.25 if conf in {None, 0.01 if is_obb else 0.001} else conf  # apply 0.25 if default val conf is passed
         no_pred = len(detections["cls"]) == 0
         if gt_cls.shape[0] == 0:  # Check if labels is empty
             if not no_pred:
@@ -380,7 +381,6 @@ class ConfusionMatrix(DataExportMixin):
         gt_classes = gt_cls.int().tolist()
         detection_classes = detections["cls"].int().tolist()
         bboxes = detections["bboxes"]
-        is_obb = bboxes.shape[1] == 5  # check if detections contains angle for OBB
         iou = batch_probiou(gt_bboxes, bboxes) if is_obb else box_iou(gt_bboxes, bboxes)
 
         x = torch.where(iou > iou_thres)
@@ -422,7 +422,7 @@ class ConfusionMatrix(DataExportMixin):
         tp = self.matrix.diagonal()  # true positives
         fp = self.matrix.sum(1) - tp  # false positives
         # fn = self.matrix.sum(0) - tp  # false negatives (missed detections)
-        return (tp[:-1], fp[:-1]) if self.task == "detect" else (tp, fp)  # remove background class if task=detect
+        return (tp, fp) if self.task == "classify" else (tp[:-1], fp[:-1])  # remove background class if task=detect
 
     @TryExcept(msg="ConfusionMatrix plot failure")
     @plt_settings()

dgenerate_ultralytics_headless-8.3.165/ultralytics/cfg/datasets/VisDrone.yaml

@@ -1,80 +0,0 @@
-# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
-
-# VisDrone2019-DET dataset https://github.com/VisDrone/VisDrone-Dataset by Tianjin University
-# Documentation: https://docs.ultralytics.com/datasets/detect/visdrone/
-# Example usage: yolo train data=VisDrone.yaml
-# parent
-# ├── ultralytics
-# └── datasets
-#     └── 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: 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
-
-# Classes
-names:
-  0: pedestrian
-  1: people
-  2: bicycle
-  3: car
-  4: van
-  5: truck
-  6: tricycle
-  7: awning-tricycle
-  8: bus
-  9: motor
-
-# Download script/URL (optional) ---------------------------------------------------------------------------------------
-download: |
-  import os
-  from pathlib import Path
-
-  from ultralytics.utils.downloads import download
-
-
-  def visdrone2yolo(dir):
-      """Convert VisDrone annotations to YOLO format, creating label files with normalized bounding box coordinates."""
-      from PIL import Image
-      from tqdm import tqdm
-
-      def convert_box(size, box):
-          # Convert VisDrone box to YOLO xywh box
-          dw = 1.0 / size[0]
-          dh = 1.0 / size[1]
-          return (box[0] + box[2] / 2) * dw, (box[1] + box[3] / 2) * dh, box[2] * dw, box[3] * dh
-
-      (dir / "labels").mkdir(parents=True, exist_ok=True)  # make labels directory
-      pbar = tqdm((dir / "annotations").glob("*.txt"), desc=f"Converting {dir}")
-      for f in pbar:
-          img_size = Image.open((dir / "images" / f.name).with_suffix(".jpg")).size
-          lines = []
-          with open(f, encoding="utf-8") as file:  # read annotation.txt
-              for row in [x.split(",") for x in file.read().strip().splitlines()]:
-                  if row[4] == "0":  # VisDrone 'ignored regions' class 0
-                      continue
-                  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")
-          
-          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
-  dir = Path(yaml["path"])  # dataset root dir
-  urls = [
-      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-train.zip",
-      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-val.zip",
-      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-test-dev.zip",
-      "https://github.com/ultralytics/assets/releases/download/v0.0.0/VisDrone2019-DET-test-challenge.zip",
-  ]
-  download(urls, dir=dir, curl=True, threads=4)
-
-  # Convert
-  for d in "VisDrone2019-DET-train", "VisDrone2019-DET-val", "VisDrone2019-DET-test-dev":
-      visdrone2yolo(dir / d)  # convert VisDrone annotations to YOLO labels