dgenerate-ultralytics-headless 8.3.137__py3-none-any.whl → 8.3.224__py3-none-any.whl

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: ../datasets/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
+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
 
 # Classes
 names:
@@ -24,9 +24,9 @@ download: |
   from pathlib import Path
 
   import numpy as np
-  import pandas as pd
-  from tqdm import tqdm
+  import polars as pl
 
+  from ultralytics.utils import TQDM
   from ultralytics.utils.downloads import download
   from ultralytics.utils.ops import xyxy2xywh
 
@@ -45,11 +45,11 @@ download: |
   # Convert labels
   names = "image", "x1", "y1", "x2", "y2", "class", "image_width", "image_height"  # column names
   for d in "annotations_train.csv", "annotations_val.csv", "annotations_test.csv":
-      x = pd.read_csv(dir / "annotations" / d, names=names).values  # annotations
+      x = pl.read_csv(dir / "annotations" / d, has_header=False, new_columns=names, infer_schema_length=None).to_numpy()  # annotations
       images, unique_images = x[:, 0], np.unique(x[:, 0])
       with open((dir / d).with_suffix(".txt").__str__().replace("annotations_", ""), "w", encoding="utf-8") as f:
           f.writelines(f"./images/{s}\n" for s in unique_images)
-      for im in tqdm(unique_images, desc=f"Converting {dir / d}"):
+      for im in TQDM(unique_images, desc=f"Converting {dir / d}"):
           cls = 0  # single-class dataset
           with open((dir / "labels" / im).with_suffix(".txt"), "a", encoding="utf-8") as f:
               for r in x[images == im]:

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: ../datasets/VOC
-train: # train images (relative to 'path')  16551 images
+path: VOC
+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
@@ -48,10 +48,8 @@ download: |
   import xml.etree.ElementTree as ET
   from pathlib import Path
 
-  from tqdm import tqdm
-
   from ultralytics.utils.downloads import download
-
+  from ultralytics.utils import ASSETS_URL, TQDM
 
   def convert_label(path, lb_path, year, image_id):
       """Converts XML annotations from VOC format to YOLO format by extracting bounding boxes and class IDs."""
@@ -81,13 +79,12 @@ download: |
 
   # Download
   dir = Path(yaml["path"])  # dataset root dir
-  url = "https://github.com/ultralytics/assets/releases/download/v0.0.0/"
   urls = [
-      f"{url}VOCtrainval_06-Nov-2007.zip",  # 446MB, 5012 images
-      f"{url}VOCtest_06-Nov-2007.zip",  # 438MB, 4953 images
-      f"{url}VOCtrainval_11-May-2012.zip",  # 1.95GB, 17126 images
+      f"{ASSETS_URL}/VOCtrainval_06-Nov-2007.zip",  # 446MB, 5012 images
+      f"{ASSETS_URL}/VOCtest_06-Nov-2007.zip",  # 438MB, 4953 images
+      f"{ASSETS_URL}/VOCtrainval_11-May-2012.zip",  # 1.95GB, 17126 images
   ]
-  download(urls, dir=dir / "images", curl=True, threads=3, exist_ok=True)  # download and unzip over existing (required)
+  download(urls, dir=dir / "images", threads=3, exist_ok=True)  # download and unzip over existing (required)
 
   # Convert
   path = dir / "images/VOCdevkit"
@@ -99,7 +96,7 @@ download: |
 
       with open(path / f"VOC{year}/ImageSets/Main/{image_set}.txt") as f:
           image_ids = f.read().strip().split()
-      for id in tqdm(image_ids, desc=f"{image_set}{year}"):
+      for id in TQDM(image_ids, desc=f"{image_set}{year}"):
           f = path / f"VOC{year}/JPEGImages/{id}.jpg"  # old img path
           lb_path = (lbs_path / f.name).with_suffix(".txt")  # new label path
           f.rename(imgs_path / f.name)  # move image

ultralytics/cfg/datasets/VisDrone.yaml

@@ -6,13 +6,13 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── VisDrone  ← downloads here (2.3 GB)
+#     └── 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
-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
+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:
@@ -31,47 +31,57 @@ names:
 download: |
   import os
   from pathlib import Path
+  import shutil
 
   from ultralytics.utils.downloads import download
+  from ultralytics.utils import ASSETS_URL, TQDM
 
 
-  def visdrone2yolo(dir):
-      """Convert VisDrone annotations to YOLO format, creating label files with normalized bounding box coordinates."""
+  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
 
-      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
+      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)
 
-      (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
+      # 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:  # read annotation.txt
+
+          with open(f, encoding="utf-8") as file:
               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")
-                  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
+                  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
+  # 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",
+      f"{ASSETS_URL}/VisDrone2019-DET-train.zip",
+      f"{ASSETS_URL}/VisDrone2019-DET-val.zip",
+      f"{ASSETS_URL}/VisDrone2019-DET-test-dev.zip",
+      # f"{ASSETS_URL}/VisDrone2019-DET-test-challenge.zip",
   ]
-  download(urls, dir=dir, curl=True, threads=4)
+  download(urls, dir=dir, 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
+  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

ultralytics/cfg/datasets/african-wildlife.yaml

@@ -6,13 +6,13 @@
 # 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: ../datasets/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
+path: african-wildlife # dataset root dir
+train: images/train # train images (relative to 'path') 1052 images
+val: images/val # val images (relative to 'path') 225 images
+test: images/test # test images (relative to 'path') 227 images
 
 # Classes
 names:

ultralytics/cfg/datasets/brain-tumor.yaml

@@ -6,13 +6,12 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── brain-tumor  ← downloads here (4.05 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: ../datasets/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')
+path: brain-tumor # dataset root dir
+train: images/train # train images (relative to 'path') 893 images
+val: images/val # val images (relative to 'path') 223 images
 
 # Classes
 names:

ultralytics/cfg/datasets/carparts-seg.yaml

@@ -6,13 +6,13 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── carparts-seg  ← downloads here (132 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: ../datasets/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
+path: carparts-seg # dataset root dir
+train: images/train # train images (relative to 'path') 3516 images
+val: images/val # val images (relative to 'path') 276 images
+test: images/test # test images (relative to 'path') 401 images
 
 # Classes
 names:

ultralytics/cfg/datasets/coco-pose.yaml

@@ -6,10 +6,10 @@
 # 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: ../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
@@ -22,16 +22,38 @@ flip_idx: [0, 2, 1, 4, 3, 6, 5, 8, 7, 10, 9, 12, 11, 14, 13, 16, 15]
 names:
   0: person
 
+# Keypoint names per class
+kpt_names:
+  0:
+    - nose
+    - left_eye
+    - right_eye
+    - left_ear
+    - right_ear
+    - left_shoulder
+    - right_shoulder
+    - left_elbow
+    - right_elbow
+    - left_wrist
+    - right_wrist
+    - left_hip
+    - right_hip
+    - left_knee
+    - right_knee
+    - left_ankle
+    - right_ankle
+
 # Download script/URL (optional)
 download: |
   from pathlib import Path
 
+  from ultralytics.utils import ASSETS_URL
   from ultralytics.utils.downloads import download
 
   # Download labels
   dir = Path(yaml["path"])  # dataset root dir
-  url = "https://github.com/ultralytics/assets/releases/download/v0.0.0/"
-  urls = [f"{url}coco2017labels-pose.zip"]
+
+  urls = [f"{ASSETS_URL}/coco2017labels-pose.zip"]
   download(urls, dir=dir.parent)
   # Download data
   urls = [

ultralytics/cfg/datasets/coco.yaml

@@ -6,10 +6,10 @@
 # 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: ../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
@@ -101,13 +101,13 @@ names:
 download: |
   from pathlib import Path
 
+  from ultralytics.utils import ASSETS_URL
   from ultralytics.utils.downloads import download
 
   # Download labels
   segments = True  # segment or box labels
   dir = Path(yaml["path"])  # dataset root dir
-  url = "https://github.com/ultralytics/assets/releases/download/v0.0.0/"
-  urls = [url + ("coco2017labels-segments.zip" if segments else "coco2017labels.zip")]  # labels
+  urls = [ASSETS_URL + ("/coco2017labels-segments.zip" if segments else "/coco2017labels.zip")]  # labels
   download(urls, dir=dir.parent)
   # Download data
   urls = [

ultralytics/cfg/datasets/coco128-seg.yaml

@@ -6,10 +6,10 @@
 # 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: ../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/cfg/datasets/coco128.yaml

@@ -6,10 +6,10 @@
 # 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: ../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/cfg/datasets/coco8-grayscale.yaml

@@ -0,0 +1,103 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+# COCO8-Grayscale dataset (first 8 images from COCO train2017) by Ultralytics
+# Documentation: https://docs.ultralytics.com/datasets/detect/coco8-grayscale/
+# Example usage: yolo train data=coco8-grayscale.yaml
+# parent
+# ├── ultralytics
+# └── datasets
+#     └── 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
+train: images/train # train images (relative to 'path') 4 images
+val: images/val # val images (relative to 'path') 4 images
+test: # test images (optional)
+
+channels: 1
+
+# Classes
+names:
+  0: person
+  1: bicycle
+  2: car
+  3: motorcycle
+  4: airplane
+  5: bus
+  6: train
+  7: truck
+  8: boat
+  9: traffic light
+  10: fire hydrant
+  11: stop sign
+  12: parking meter
+  13: bench
+  14: bird
+  15: cat
+  16: dog
+  17: horse
+  18: sheep
+  19: cow
+  20: elephant
+  21: bear
+  22: zebra
+  23: giraffe
+  24: backpack
+  25: umbrella
+  26: handbag
+  27: tie
+  28: suitcase
+  29: frisbee
+  30: skis
+  31: snowboard
+  32: sports ball
+  33: kite
+  34: baseball bat
+  35: baseball glove
+  36: skateboard
+  37: surfboard
+  38: tennis racket
+  39: bottle
+  40: wine glass
+  41: cup
+  42: fork
+  43: knife
+  44: spoon
+  45: bowl
+  46: banana
+  47: apple
+  48: sandwich
+  49: orange
+  50: broccoli
+  51: carrot
+  52: hot dog
+  53: pizza
+  54: donut
+  55: cake
+  56: chair
+  57: couch
+  58: potted plant
+  59: bed
+  60: dining table
+  61: toilet
+  62: tv
+  63: laptop
+  64: mouse
+  65: remote
+  66: keyboard
+  67: cell phone
+  68: microwave
+  69: oven
+  70: toaster
+  71: sink
+  72: refrigerator
+  73: book
+  74: clock
+  75: vase
+  76: scissors
+  77: teddy bear
+  78: hair drier
+  79: toothbrush
+
+# Download script/URL (optional)
+download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8-grayscale.zip

ultralytics/cfg/datasets/coco8-multispectral.yaml

@@ -6,10 +6,10 @@
 # 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: ../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/cfg/datasets/coco8-pose.yaml

@@ -6,10 +6,10 @@
 # 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: ../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)
@@ -22,5 +22,26 @@ flip_idx: [0, 2, 1, 4, 3, 6, 5, 8, 7, 10, 9, 12, 11, 14, 13, 16, 15]
 names:
   0: person
 
+# Keypoint names per class
+kpt_names:
+  0:
+    - nose
+    - left_eye
+    - right_eye
+    - left_ear
+    - right_ear
+    - left_shoulder
+    - right_shoulder
+    - left_elbow
+    - right_elbow
+    - left_wrist
+    - right_wrist
+    - left_hip
+    - right_hip
+    - left_knee
+    - right_knee
+    - left_ankle
+    - right_ankle
+
 # Download script/URL (optional)
 download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8-pose.zip

ultralytics/cfg/datasets/coco8-seg.yaml

@@ -6,10 +6,10 @@
 # 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: ../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/cfg/datasets/coco8.yaml

@@ -6,10 +6,10 @@
 # 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: ../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/cfg/datasets/construction-ppe.yaml

@@ -0,0 +1,32 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+# Construction-PPE dataset by Ultralytics
+# Documentation: https://docs.ultralytics.com/datasets/detect/construction-ppe/
+# Example usage: yolo train data=construction-ppe.yaml
+# parent
+# ├── ultralytics
+# └── datasets
+#     └── construction-ppe ← downloads here (178.4 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: construction-ppe # dataset root dir
+train: images/train # train images (relative to 'path') 1132 images
+val: images/val # val images (relative to 'path') 143 images
+test: images/test # test images (relative to 'path') 141 images
+
+# Classes
+names:
+  0: helmet
+  1: gloves
+  2: vest
+  3: boots
+  4: goggles
+  5: none
+  6: Person
+  7: no_helmet
+  8: no_goggle
+  9: no_gloves
+  10: no_boots
+
+# Download script/URL (optional)
+download: https://github.com/ultralytics/assets/releases/download/v0.0.0/construction-ppe.zip

ultralytics/cfg/datasets/crack-seg.yaml

@@ -6,13 +6,13 @@
 # parent
 # ├── ultralytics
 # └── datasets
-#     └── crack-seg  ← downloads here (91.2 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: ../datasets/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
+path: crack-seg # dataset root dir
+train: images/train # train images (relative to 'path') 3717 images
+val: images/val # val images (relative to 'path') 112 images
+test: images/test # test images (relative to 'path') 200 images
 
 # Classes
 names: