dgenerate-ultralytics-headless 8.3.243__tar.gz → 8.3.245__tar.gz

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.243
+Version: 8.3.245
 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.243 → dgenerate_ultralytics_headless-8.3.245}/dgenerate_ultralytics_headless.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.243
+Version: 8.3.245
 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.243 → dgenerate_ultralytics_headless-8.3.245}/tests/test_engine.py

@@ -13,7 +13,7 @@ from ultralytics.models.yolo import classify, detect, segment
 from ultralytics.utils import ASSETS, DEFAULT_CFG, WEIGHTS_DIR
 
 
-def test_func(*args):
+def test_func(*args, **kwargs):
     """Test function callback for evaluating YOLO model performance metrics."""
     print("callback test passed")
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/__init__.py

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

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov8n-cls.yaml' will call yolov8-cls.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=rtdetr-l.yaml' will call rtdetr-l.yaml with scale 'l'
   # [depth, width, max_channels]
   l: [1.00, 1.00, 1024]
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov8n-cls.yaml' will call yolov8-cls.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=rtdetr-resnet101.yaml' will call rtdetr-resnet101.yaml with scale 'l'
   # [depth, width, max_channels]
   l: [1.00, 1.00, 1024]
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov8n-cls.yaml' will call yolov8-cls.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=rtdetr-resnet50.yaml' will call rtdetr-resnet50.yaml with scale 'l'
   # [depth, width, max_channels]
   l: [1.00, 1.00, 1024]
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/rt-detr/rtdetr-x.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov8n-cls.yaml' will call yolov8-cls.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=rtdetr-x.yaml' will call rtdetr-x.yaml with scale 'x'
   # [depth, width, max_channels]
   x: [1.00, 1.00, 2048]
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10b.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   b: [0.67, 1.00, 512]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10l.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   l: [1.00, 1.00, 512]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10m.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   m: [0.67, 0.75, 768]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10n.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   n: [0.33, 0.25, 1024]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10s.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   s: [0.33, 0.50, 1024]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v10/yolov10x.yaml

@@ -6,7 +6,7 @@
 
 # Parameters
 nc: 80 # number of classes
-scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' will call yolov10.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov10n.yaml' uses the 'n' scale
   # [depth, width, max_channels]
   x: [1.00, 1.25, 512]
 
@@ -24,7 +24,7 @@ backbone:
   - [-1, 1, SPPF, [1024, 5]] # 9
   - [-1, 1, PSA, [1024]] # 10
 
-# YOLOv10.0n head
+# YOLOv10 head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]] # cat backbone P4

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v3/yolov3-tiny.yaml

@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-# Ultralytics YOLOv3-tiiny object detection model with P4/16 - P5/32 outputs
+# Ultralytics YOLOv3-tiny object detection model with P4/16 - P5/32 outputs
 # Model docs: https://docs.ultralytics.com/models/yolov3
 # Task docs: https://docs.ultralytics.com/tasks/detect
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v6/yolov6.yaml

@@ -7,7 +7,7 @@
 # Parameters
 nc: 80 # number of classes
 activation: torch.nn.ReLU() # (optional) model default activation function
-scales: # model compound scaling constants, i.e. 'model=yolov6n.yaml' will call yolov8.yaml with scale 'n'
+scales: # model compound scaling constants, i.e. 'model=yolov6n.yaml' will call yolov6.yaml with scale 'n'
   # [depth, width, max_channels]
   n: [0.33, 0.25, 1024]
   s: [0.33, 0.50, 1024]

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v8/yolov8-seg-p6.yaml

@@ -56,4 +56,4 @@ head:
   - [[-1, 11], 1, Concat, [1]] # cat head P6
   - [-1, 3, C2, [1024, False]] # 29 (P6/64-xlarge)
 
-  - [[20, 23, 26, 29], 1, Segment, [nc, 32, 256]] # Pose(P3, P4, P5, P6)
+  - [[20, 23, 26, 29], 1, Segment, [nc, 32, 256]] # Segment(P3, P4, P5, P6)

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/cfg/models/v9/yolov9s.yaml

@@ -38,4 +38,4 @@ head:
   - [[-1, 9], 1, Concat, [1]] # cat head P5
   - [-1, 1, RepNCSPELAN4, [256, 256, 128, 3]] # 21 (P5/32-large)
 
-  - [[15, 18, 21], 1, Detect, [nc]] # Detect(P3, P4 P5)
+  - [[15, 18, 21], 1, Detect, [nc]] # Detect(P3, P4, P5)

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/engine/exporter.py

@@ -87,6 +87,7 @@ from ultralytics.utils import (
     IS_COLAB,
     IS_DEBIAN_BOOKWORM,
     IS_DEBIAN_TRIXIE,
+    IS_DOCKER,
     IS_JETSON,
     IS_RASPBERRYPI,
     IS_UBUNTU,
@@ -1187,10 +1188,11 @@ class Exporter:
         """
         LOGGER.info(f"\n{prefix} starting export with ExecuTorch...")
         assert TORCH_2_9, f"ExecuTorch export requires torch>=2.9.0 but torch=={TORCH_VERSION} is installed"
-        # TorchAO release compatibility table bug https://github.com/pytorch/ao/issues/2919
-        # Setuptools bug: https://github.com/pypa/setuptools/issues/4483
-        check_requirements("setuptools<71.0.0")  # Setuptools bug: https://github.com/pypa/setuptools/issues/4483
-        check_requirements(("executorch==1.0.1", "flatbuffers"))
+
+        # BUG executorch build on arm64 Docker requires packaging>=22.0 https://github.com/pypa/setuptools/issues/4483
+        if LINUX and ARM64 and IS_DOCKER:
+            check_requirements("packaging>=22.0")
+        check_requirements("executorch==1.0.1", "flatbuffers")
         # Pin numpy to avoid coremltools errors with numpy>=2.4.0, must be separate
         check_requirements("numpy<=2.3.5")
 

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/engine/model.py

@@ -523,7 +523,7 @@ class Model(torch.nn.Module):
         args = {**self.overrides, **custom, **kwargs}  # highest priority args on the right
         prompts = args.pop("prompts", None)  # for SAM-type models
 
-        if not self.predictor:
+        if not self.predictor or self.predictor.args.device != args.get("device", self.predictor.args.device):
             self.predictor = (predictor or self._smart_load("predictor"))(overrides=args, _callbacks=self.callbacks)
             self.predictor.setup_model(model=self.model, verbose=is_cli)
         else:  # only update args if predictor is already setup

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/nn/modules/head.py

@@ -1078,9 +1078,9 @@ class RTDETRDecoder(nn.Module):
         enc_outputs_scores = self.enc_score_head(features)  # (bs, h*w, nc)
 
         # Query selection
-        # (bs, num_queries)
+        # (bs*num_queries,)
         topk_ind = torch.topk(enc_outputs_scores.max(-1).values, self.num_queries, dim=1).indices.view(-1)
-        # (bs, num_queries)
+        # (bs*num_queries,)
         batch_ind = torch.arange(end=bs, dtype=topk_ind.dtype).unsqueeze(-1).repeat(1, self.num_queries).view(-1)
 
         # (bs, num_queries, 256)

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/nn/modules/utils.py

@@ -147,7 +147,7 @@ def multi_scale_deformable_attn_pytorch(
         sampling_value_list.append(sampling_value_l_)
     # (bs, num_queries, num_heads, num_levels, num_points) ->
     # (bs, num_heads, num_queries, num_levels, num_points) ->
-    # (bs, num_heads, 1, num_queries, num_levels*num_points)
+    # (bs*num_heads, 1, num_queries, num_levels*num_points)
     attention_weights = attention_weights.transpose(1, 2).reshape(
         bs * num_heads, 1, num_queries, num_levels * num_points
     )

{dgenerate_ultralytics_headless-8.3.243 → dgenerate_ultralytics_headless-8.3.245}/ultralytics/utils/callbacks/platform.py

@@ -90,11 +90,18 @@ def _upload_model_async(model_path, project, name):
 
 def _get_environment_info():
     """Collect comprehensive environment info using existing ultralytics utilities."""
+    import shutil
+
+    import psutil
     import torch
 
     from ultralytics import __version__
     from ultralytics.utils.torch_utils import get_cpu_info, get_gpu_info
 
+    # Get RAM and disk totals
+    memory = psutil.virtual_memory()
+    disk_usage = shutil.disk_usage("/")
+
     env = {
         "ultralyticsVersion": __version__,
         "hostname": socket.gethostname(),
@@ -105,6 +112,8 @@ def _get_environment_info():
         "cpuCount": os.cpu_count() or 0,
         "cpu": get_cpu_info(),
         "command": " ".join(sys.argv),
+        "totalRamGb": round(memory.total / (1 << 30), 1),  # Total RAM in GB
+        "totalDiskGb": round(disk_usage.total / (1 << 30), 1),  # Total disk in GB
     }
 
     # Git info using cached GIT singleton (no subprocess calls)
@@ -258,7 +267,9 @@ def on_train_end(trainer):
 
     # Upload best model (blocking to ensure it completes)
     model_path = None
+    model_size = None
     if trainer.best and Path(trainer.best).exists():
+        model_size = Path(trainer.best).stat().st_size
         model_path = _upload_model(trainer.best, project, name)
 
     # Send training complete
@@ -270,6 +281,7 @@ def on_train_end(trainer):
                 "bestEpoch": getattr(trainer, "best_epoch", trainer.epoch),
                 "bestFitness": trainer.best_fitness,
                 "modelPath": model_path or str(trainer.best) if trainer.best else None,
+                "modelSize": model_size,
             }
         },
         project,