dgenerate-ultralytics-headless 8.3.249__py3-none-any.whl → 8.3.250__py3-none-any.whl

{dgenerate_ultralytics_headless-8.3.249.dist-info → dgenerate_ultralytics_headless-8.3.250.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.249
+Version: 8.3.250
 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.249.dist-info → dgenerate_ultralytics_headless-8.3.250.dist-info}/RECORD

@@ -1,4 +1,4 @@
-dgenerate_ultralytics_headless-8.3.249.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
+dgenerate_ultralytics_headless-8.3.250.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
 tests/__init__.py,sha256=bCox_hLdGRFYGLb2kd722VdNP2zEXNYNuLLYtqZSrbw,804
 tests/conftest.py,sha256=mOy9lGpNp7lk1hHl6_pVE0f9cU-72gnkoSm4TO-CNZU,2318
 tests/test_cli.py,sha256=GhIFHi-_WIJpDgoGNRi0DnjbfwP1wHbklBMnkCM-P_4,5464
@@ -8,11 +8,11 @@ tests/test_exports.py,sha256=5G5EgDmars6d-N7TVnJdDFWId0IJs-yw03DvdQIjrNU,14246
 tests/test_integrations.py,sha256=6QgSh9n0J04RdUYz08VeVOnKmf4S5MDEQ0chzS7jo_c,6220
 tests/test_python.py,sha256=viMvRajIbDZdm64hRRg9i8qZ1sU9frwB69e56mxwEXk,29266
 tests/test_solutions.py,sha256=CIaphpmOXgz9AE9xcm1RWODKrwGfZLCc84IggGXArNM,14122
-ultralytics/__init__.py,sha256=nbR9xzuCbfetnkDGSTE5sxiBO0svWjIEk40CpLGaTJo,1302
+ultralytics/__init__.py,sha256=v-hVhlejM2hHyVyOWQ59-HPirtK5Go1IZS0C8EMKfKY,1302
 ultralytics/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
-ultralytics/cfg/__init__.py,sha256=msy7NE7le831rv5-eRdLFkHRkf8o7m07HRkY4810n9s,40208
+ultralytics/cfg/__init__.py,sha256=T5bv7NdySatnqEvtOW4_qojIC3pmAbp0GlCwyk0RazU,40368
 ultralytics/cfg/default.yaml,sha256=KKENSHolDSto1HJVGjBvTXvz9ae-XMcYRzKrjU3QfZc,8912
 ultralytics/cfg/datasets/Argoverse.yaml,sha256=QGpdh3Hj5dFrvbsaE_8rAVj9BO4XpKTB7uhXaTTnE-o,3364
 ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=KE7VC-ZMDSei1pLPm-pdk_ZAMRU_gLwGgtIQNbwp6dA,1212
@@ -22,6 +22,7 @@ ultralytics/cfg/datasets/HomeObjects-3K.yaml,sha256=xEtSqEad-rtfGuIrERjjhdISggmP
 ultralytics/cfg/datasets/ImageNet.yaml,sha256=N9NHhIgnlNIBqZZbzQZAW3aCnz6RSXQABnopaDs5BmE,42529
 ultralytics/cfg/datasets/Objects365.yaml,sha256=8Bl-NAm0mlMW8EfMsz39JZo-HCvmp0ejJXaMeoHTpqw,9649
 ultralytics/cfg/datasets/SKU-110K.yaml,sha256=xvRkq3SdDOwBA91U85bln7HTXkod5MvFX6pt1PxTjJE,2609
+ultralytics/cfg/datasets/TT100K.yaml,sha256=me7mEq3qDeL9_DpQBRS9jCVxsuByKuDBOwB0ePX0_LE,7369
 ultralytics/cfg/datasets/VOC.yaml,sha256=XpaegRHjp7xZnenOuA9zgg2lQURSL-o7mLQwzIKKuqM,3803
 ultralytics/cfg/datasets/VisDrone.yaml,sha256=PfudojW5av_5q-dC9VsG_xhvuv9cTGEpRp4loXCJ4Ng,3397
 ultralytics/cfg/datasets/african-wildlife.yaml,sha256=6UfO_gnwJEDVq05p72IMJfkTIKZlXKNLSeKru-JyTrQ,915
@@ -127,7 +128,7 @@ ultralytics/engine/exporter.py,sha256=Ncf5GK5xAqSu0DH-6z5V53qZB7LstDJFTMF5a-7VQf
 ultralytics/engine/model.py,sha256=61ea1rB0wmL0CCaEr8p5gzneH0eL55OOMaTcFt8fR80,53079
 ultralytics/engine/predictor.py,sha256=neYmNDX27Vv3ggk9xqaKlH6XzB2vlFIghU5o7ZC0zFo,22838
 ultralytics/engine/results.py,sha256=DomI01voqR_i7v8LhDGb6jWCprWB4H6I436GSO2NMBY,68030
-ultralytics/engine/trainer.py,sha256=mqVrhL8xnJwwKJVjxDEiiwu0WH48Ne5dB4SXxlxyHh4,45479
+ultralytics/engine/trainer.py,sha256=UQfxbcT983v9bXYNYH5MvzZmuega0ZUW-KqUoq4vxjE,45798
 ultralytics/engine/tuner.py,sha256=qiozSxYC-Hk1TQgyftrYTKkqLrrwFzjjkT6mOYR3Vjc,21460
 ultralytics/engine/validator.py,sha256=2rqdVt4hB9ruMJq-L7PbaCNFwuERS7ZHdVSg91RM3wk,17761
 ultralytics/hub/__init__.py,sha256=Z0K_E00jzQh90b18q3IDChwVmTvyIYp6C00sCV-n2F8,6709
@@ -265,7 +266,7 @@ ultralytics/utils/git.py,sha256=UdqeIiiEzg1qkerAZrg5YtTYPuJYwrpxW9N_6Pq6s8U,5501
 ultralytics/utils/instance.py,sha256=11mhefvTI9ftMqSirXuiViAi0Fxlo6v84qvNxfRNUoE,18862
 ultralytics/utils/logger.py,sha256=T5iaNnaqbCvx_FZf1dhVkr5FVxyxb4vO17t4SJfCIhg,19132
 ultralytics/utils/loss.py,sha256=t-z7qkvqF8OtuRHrj2wmvClZV2CCumIRi9jnqkc9i_A,39573
-ultralytics/utils/metrics.py,sha256=dpS9jSPf3dqozcrkiraKhYBI03U2t-_lt8pWNCijGww,69152
+ultralytics/utils/metrics.py,sha256=SpyMGnuRwwmorJqSdUsDQquVpGmgfj1X3PNDiw_ZZWM,69152
 ultralytics/utils/nms.py,sha256=zv1rOzMF6WU8Kdk41VzNf1H1EMt_vZHcbDFbg3mnN2o,14248
 ultralytics/utils/ops.py,sha256=nWvTLJSBeW_XrxCy5Ytxl7sZJHp2sRqyCv4mm8QwYnw,25797
 ultralytics/utils/patches.py,sha256=mD3slAMAhcezzP42_fOWmacNMU6zXB68Br4_EBCyIjs,7117
@@ -283,7 +284,7 @@ ultralytics/utils/callbacks/dvc.py,sha256=YT0Sa5P8Huj8Fn9jM2P6MYzUY3PIVxsa5BInVi
 ultralytics/utils/callbacks/hub.py,sha256=fVLqqr3ZM6hoYFlVMEeejfq1MWDrkWCskPFOG3HGILQ,4159
 ultralytics/utils/callbacks/mlflow.py,sha256=wCXjQgdufp9LYujqMzLZOmIOur6kvrApHNeo9dA7t_g,5323
 ultralytics/utils/callbacks/neptune.py,sha256=_vt3cMwDHCR-LyT3KtRikGpj6AG11oQ-skUUUUdZ74o,4391
-ultralytics/utils/callbacks/platform.py,sha256=EbKwGV0kVX1ZfwR6MBOPSeDKWpd4-nVO14uydMpJlRs,14798
+ultralytics/utils/callbacks/platform.py,sha256=R9KcdNQUKs-mS_oOxXeKK51uYWlGhVkvQnarKO6qmOk,14804
 ultralytics/utils/callbacks/raytune.py,sha256=Y0dFyNZVRuFovSh7nkgUIHTQL3xIXOACElgHuYbg_5I,1278
 ultralytics/utils/callbacks/tensorboard.py,sha256=PTJYvD2gqRUN8xw5VoTjvKnu2adukLfvhMlDgTnTiFU,4952
 ultralytics/utils/callbacks/wb.py,sha256=ghmL3gigOa-z_F54-TzMraKw9MAaYX-Wk4H8dLoRvX8,7705
@@ -291,8 +292,8 @@ ultralytics/utils/export/__init__.py,sha256=Cfh-PwVfTF_lwPp-Ss4wiX4z8Sm1XRPklsqd
 ultralytics/utils/export/engine.py,sha256=23-lC6dNsmz5vprSJzaN7UGNXrFlVedNcqhlOH_IXes,9956
 ultralytics/utils/export/imx.py,sha256=2_mcNzqRIk5LB92JofqNYLN0kkQke1UgKT2jWmEy_l4,13300
 ultralytics/utils/export/tensorflow.py,sha256=igYzwbdblb9YgfV4Jgl5lMvynuVRcF51dAzI7j-BBI0,9966
-dgenerate_ultralytics_headless-8.3.249.dist-info/METADATA,sha256=msxmb2FNZ3CXXLbjpCua4wN7TWRuuvQA8wtdINrhQ4k,38799
-dgenerate_ultralytics_headless-8.3.249.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dgenerate_ultralytics_headless-8.3.249.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
-dgenerate_ultralytics_headless-8.3.249.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-dgenerate_ultralytics_headless-8.3.249.dist-info/RECORD,,
+dgenerate_ultralytics_headless-8.3.250.dist-info/METADATA,sha256=iK_C4dYeMXmNoXQZhkhEq_uxDjF4SDef5-gWVXAZEgc,38799
+dgenerate_ultralytics_headless-8.3.250.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dgenerate_ultralytics_headless-8.3.250.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
+dgenerate_ultralytics_headless-8.3.250.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+dgenerate_ultralytics_headless-8.3.250.dist-info/RECORD,,

ultralytics/__init__.py

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

ultralytics/cfg/__init__.py

@@ -410,7 +410,9 @@ def get_save_dir(args: SimpleNamespace, name: str | None = None) -> Path:
     else:
         from ultralytics.utils.files import increment_path
 
-        project = args.project or (ROOT.parent / "tests/tmp/runs" if TESTS_RUNNING else RUNS_DIR) / args.task
+        runs = (ROOT.parent / "tests/tmp/runs" if TESTS_RUNNING else RUNS_DIR) / args.task
+        nested = args.project and len(Path(args.project).parts) > 1  # e.g. "user/project" or "org\repo"
+        project = runs / args.project if nested else args.project or runs
         name = name or args.name or f"{args.mode}"
         save_dir = increment_path(Path(project) / name, exist_ok=args.exist_ok if RANK in {-1, 0} else True)
 

ultralytics/cfg/datasets/TT100K.yaml

@@ -0,0 +1,356 @@
+# Ultralytics YOLO 🚀 AGPL-3.0 License - https://ultralytics.com/license
+# Tsinghua-Tencent 100K Traffic Sign Detection Dataset
+# Dataset: https://cg.cs.tsinghua.edu.cn/traffic-sign/
+# Paper: Traffic-Sign Detection and Classification in the Wild (CVPR 2016)
+# Documentation: 100,000 images with 30,000+ traffic sign annotations
+# Example usage: yolo train data=TT100K.yaml
+# parent
+# ├── ultralytics
+# └── datasets
+#     └── TT100K  ← downloads here (~30 GB with images)
+
+# Train/val/test sets
+path: TT100K # dataset root dir
+train: images/train # train images (relative to 'path')
+val: images/val # val images (relative to 'path')
+test: images/test # test images (relative to 'path')
+
+# Classes
+nc: 221
+names:
+  0: pl5
+  1: pl10
+  2: pl15
+  3: pl20
+  4: pl25
+  5: pl30
+  6: pl40
+  7: pl50
+  8: pl60
+  9: pl70
+  10: pl80
+  11: pl90
+  12: pl100
+  13: pl110
+  14: pl120
+  15: pm5
+  16: pm10
+  17: pm13
+  18: pm15
+  19: pm20
+  20: pm25
+  21: pm30
+  22: pm35
+  23: pm40
+  24: pm46
+  25: pm50
+  26: pm55
+  27: pm8
+  28: pn
+  29: pne
+  30: ph4
+  31: ph4.5
+  32: ph5
+  33: ps
+  34: pg
+  35: ph1.5
+  36: ph2
+  37: ph2.1
+  38: ph2.2
+  39: ph2.4
+  40: ph2.5
+  41: ph2.8
+  42: ph2.9
+  43: ph3
+  44: ph3.2
+  45: ph3.5
+  46: ph3.8
+  47: ph4.2
+  48: ph4.3
+  49: ph4.8
+  50: ph5.3
+  51: ph5.5
+  52: pb
+  53: pr10
+  54: pr100
+  55: pr20
+  56: pr30
+  57: pr40
+  58: pr45
+  59: pr50
+  60: pr60
+  61: pr70
+  62: pr80
+  63: pr90
+  64: p1
+  65: p2
+  66: p3
+  67: p4
+  68: p5
+  69: p6
+  70: p7
+  71: p8
+  72: p9
+  73: p10
+  74: p11
+  75: p12
+  76: p13
+  77: p14
+  78: p15
+  79: p16
+  80: p17
+  81: p18
+  82: p19
+  83: p20
+  84: p21
+  85: p22
+  86: p23
+  87: p24
+  88: p25
+  89: p26
+  90: p27
+  91: p28
+  92: pa8
+  93: pa10
+  94: pa12
+  95: pa13
+  96: pa14
+  97: pb5
+  98: pc
+  99: pg
+  100: ph1
+  101: ph1.3
+  102: ph1.5
+  103: ph2
+  104: ph3
+  105: ph4
+  106: ph5
+  107: pi
+  108: pl0
+  109: pl4
+  110: pl5
+  111: pl8
+  112: pl10
+  113: pl15
+  114: pl20
+  115: pl25
+  116: pl30
+  117: pl35
+  118: pl40
+  119: pl50
+  120: pl60
+  121: pl65
+  122: pl70
+  123: pl80
+  124: pl90
+  125: pl100
+  126: pl110
+  127: pl120
+  128: pm2
+  129: pm8
+  130: pm10
+  131: pm13
+  132: pm15
+  133: pm20
+  134: pm25
+  135: pm30
+  136: pm35
+  137: pm40
+  138: pm46
+  139: pm50
+  140: pm55
+  141: pn
+  142: pne
+  143: po
+  144: pr10
+  145: pr100
+  146: pr20
+  147: pr30
+  148: pr40
+  149: pr45
+  150: pr50
+  151: pr60
+  152: pr70
+  153: pr80
+  154: ps
+  155: w1
+  156: w2
+  157: w3
+  158: w5
+  159: w8
+  160: w10
+  161: w12
+  162: w13
+  163: w16
+  164: w18
+  165: w20
+  166: w21
+  167: w22
+  168: w24
+  169: w28
+  170: w30
+  171: w31
+  172: w32
+  173: w34
+  174: w35
+  175: w37
+  176: w38
+  177: w41
+  178: w42
+  179: w43
+  180: w44
+  181: w45
+  182: w46
+  183: w47
+  184: w48
+  185: w49
+  186: w50
+  187: w51
+  188: w52
+  189: w53
+  190: w54
+  191: w55
+  192: w56
+  193: w57
+  194: w58
+  195: w59
+  196: w60
+  197: w62
+  198: w63
+  199: w66
+  200: i1
+  201: i2
+  202: i3
+  203: i4
+  204: i5
+  205: i6
+  206: i7
+  207: i8
+  208: i9
+  209: i10
+  210: i11
+  211: i12
+  212: i13
+  213: i14
+  214: i15
+  215: il60
+  216: il80
+  217: il100
+  218: il110
+  219: io
+  220: ip
+
+# Download script/URL (optional) ---------------------------------------------------------------------------------------
+download: |
+  import os
+  import json
+  import shutil
+  from pathlib import Path
+  from ultralytics.utils.downloads import download
+  from ultralytics.utils import TQDM
+
+  def tt100k2yolo(dir):
+      """Convert TT100K annotations to YOLO format."""
+      from PIL import Image
+      
+      data_dir = dir / "data"
+      anno_file = data_dir / "annotations.json"
+      
+      print("Loading annotations...")
+      with open(anno_file, 'r', encoding='utf-8') as f:
+          data = json.load(f)
+      
+      # Get all unique classes
+      classes = set()
+      for img_id, img_data in data['imgs'].items():
+          for obj in img_data.get('objects', []):
+              classes.add(obj['category'])
+      class_to_idx = {cls: idx for idx, cls in enumerate(yaml["names"]) if cls in classes}
+      class_to_idx = {cls: idx for idx, cls in enumerate(classes)}
+      
+      print(f"Found {len(classes)} traffic sign classes")
+      
+      # Create directories
+      for split in ['train', 'val', 'test']:
+          (dir / 'images' / split).mkdir(parents=True, exist_ok=True)
+          (dir / 'labels' / split).mkdir(parents=True, exist_ok=True)
+      
+      print("Converting annotations to YOLO format...")
+      for img_id, img_data in TQDM(data['imgs'].items(), desc="Processing"):
+          img_path_str = img_data['path']
+          if 'train' in img_path_str:
+              split = 'train'
+          elif 'test' in img_path_str:
+              split = 'test'
+          else:
+              split = 'val'
+          
+          # Source and destination paths
+          src_img = data_dir / img_path_str
+          if not src_img.exists():
+              continue
+          
+          img_name = src_img.name
+          dst_img = dir / 'images' / split / img_name
+          
+          # Get image dimensions
+          try:
+              with Image.open(src_img) as img:
+                  img_width, img_height = img.size
+          except Exception as e:
+              print(f"Error reading {src_img}: {e}")
+              continue
+          shutil.copy2(src_img, dst_img)
+          label_file = dir / 'labels' / split / f"{src_img.stem}.txt"
+          lines = []
+          
+          for obj in img_data.get('objects', []):
+              category = obj['category']
+              if category not in class_to_idx:
+                  continue
+              
+              bbox = obj['bbox']
+              xmin, ymin = bbox['xmin'], bbox['ymin']
+              xmax, ymax = bbox['xmax'], bbox['ymax']
+              
+              # Convert to YOLO format (normalized center coordinates and dimensions)
+              x_center = ((xmin + xmax) / 2.0) / img_width
+              y_center = ((ymin + ymax) / 2.0) / img_height
+              width = (xmax - xmin) / img_width
+              height = (ymax - ymin) / img_height
+              
+              # Ensure valid coordinates
+              x_center = max(0, min(1, x_center))
+              y_center = max(0, min(1, y_center))
+              width = max(0, min(1, width))
+              height = max(0, min(1, height))
+              
+              cls_idx = class_to_idx[category]
+              lines.append(f"{cls_idx} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}\n")
+          
+          # Write label file
+          if lines:
+              label_file.write_text("".join(lines), encoding="utf-8")
+      
+      print(f"Conversion complete!")
+      print(f"Found {len(classes)} classes: {classes[:10]}..." if len(classes) > 10 else f"Classes: {classes}")
+      
+      return classes
+
+  # Download dataset
+  dir = Path(yaml['path'])  # dataset root dir
+
+  # TT100K dataset URLs
+  urls = [
+      'https://cg.cs.tsinghua.edu.cn/traffic-sign/data_model_code/data.zip',  # Main dataset with annotations
+  ]
+
+  print("Downloading TT100K dataset...")
+  print("Note: This dataset is large (~30GB). Download may take some time.")
+  print("Dataset is under CC-BY-NC license for non-commercial use only.")
+
+  # Download and extract
+  download(urls, dir=dir, unzip=True, delete=False, curl=False, threads=1)
+
+  # Convert dataset in YOLO format
+  classes = tt100k2yolo(dir)

ultralytics/engine/trainer.py

@@ -631,13 +631,17 @@ class BaseTrainer:
         try:
             if self.args.task == "classify":
                 data = check_cls_dataset(self.args.data)
-            elif str(self.args.data).rsplit(".", 1)[-1] == "ndjson":
-                # Convert NDJSON to YOLO format
+            elif str(self.args.data).rsplit(".", 1)[-1] == "ndjson" or (
+                str(self.args.data).startswith("ul://") and "/datasets/" in str(self.args.data)
+            ):
+                # Convert NDJSON to YOLO format (including ul:// platform dataset URIs)
                 import asyncio
 
                 from ultralytics.data.converter import convert_ndjson_to_yolo
+                from ultralytics.utils.checks import check_file
 
-                yaml_path = asyncio.run(convert_ndjson_to_yolo(self.args.data))
+                ndjson_file = check_file(self.args.data)  # Resolve ul:// or URL to local .ndjson file
+                yaml_path = asyncio.run(convert_ndjson_to_yolo(ndjson_file))
                 self.args.data = str(yaml_path)
                 data = check_det_dataset(self.args.data)
             elif str(self.args.data).rsplit(".", 1)[-1] in {"yaml", "yml"} or self.args.task in {

ultralytics/utils/callbacks/platform.py

@@ -62,7 +62,7 @@ def resolve_platform_uri(uri, hard=True):
     if not api_key:
         raise ValueError(f"ULTRALYTICS_API_KEY required for '{uri}'. Get key at https://alpha.ultralytics.com/settings")
 
-    base = "https://alpha.ultralytics.com/api/v1"
+    base = "https://alpha.ultralytics.com/api/webhooks"
     headers = {"Authorization": f"Bearer {api_key}"}
 
     # ul://username/datasets/slug

ultralytics/utils/metrics.py

@@ -315,7 +315,7 @@ class ConfusionMatrix(DataExportMixin):
         matches (dict): Contains the indices of ground truths and predictions categorized into TP, FP and FN.
     """
 
-    def __init__(self, names: dict[int, str] = [], task: str = "detect", save_matches: bool = False):
+    def __init__(self, names: dict[int, str] = {}, task: str = "detect", save_matches: bool = False):
         """Initialize a ConfusionMatrix instance.
 
         Args: