ultralytics-opencv-headless 8.3.251__tar.gz → 8.3.253__tar.gz

{ultralytics_opencv_headless-8.3.251/ultralytics_opencv_headless.egg-info → ultralytics_opencv_headless-8.3.253}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-opencv-headless
-Version: 8.3.251
+Version: 8.3.253
 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_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/__init__.py

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

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/engine/exporter.py

@@ -505,6 +505,7 @@ class Exporter:
                 m.format = self.args.format
                 m.max_det = self.args.max_det
                 m.xyxy = self.args.nms and not coreml
+                m.shape = None  # reset cached shape for new export input size
                 if hasattr(model, "pe") and hasattr(m, "fuse"):  # for YOLOE models
                     m.fuse(model.pe.to(self.device))
             elif isinstance(m, C2f) and not is_tf_format:
@@ -812,11 +813,11 @@ class Exporter:
         assert not IS_JETSON, "Jetson Paddle exports not supported yet"
         check_requirements(
             (
-                "paddlepaddle-gpu"
+                "paddlepaddle-gpu>=3.0.0,!=3.3.0"  # exclude 3.3.0 https://github.com/PaddlePaddle/Paddle/issues/77340
                 if torch.cuda.is_available()
                 else "paddlepaddle==3.0.0"  # pin 3.0.0 for ARM64
                 if ARM64
-                else "paddlepaddle>=3.0.0",
+                else "paddlepaddle>=3.0.0,!=3.3.0",  # exclude 3.3.0 https://github.com/PaddlePaddle/Paddle/issues/77340
                 "x2paddle",
             )
         )
@@ -859,7 +860,8 @@ class Exporter:
     @try_export
     def export_ncnn(self, prefix=colorstr("NCNN:")):
         """Export YOLO model to NCNN format using PNNX https://github.com/pnnx/pnnx."""
-        check_requirements("ncnn", cmds="--no-deps")  # no deps to avoid installing opencv-python
+        # use git source for ARM64 due to broken PyPI packages https://github.com/Tencent/ncnn/issues/6509
+        check_requirements("git+https://github.com/Tencent/ncnn.git" if ARM64 else "ncnn", cmds="--no-deps")
         check_requirements("pnnx")
         import ncnn
         import pnnx

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/nn/autobackend.py

@@ -497,11 +497,11 @@ class AutoBackend(nn.Module):
         elif paddle:
             LOGGER.info(f"Loading {w} for PaddlePaddle inference...")
             check_requirements(
-                "paddlepaddle-gpu"
+                "paddlepaddle-gpu>=3.0.0,!=3.3.0"  # exclude 3.3.0 https://github.com/PaddlePaddle/Paddle/issues/77340
                 if torch.cuda.is_available()
                 else "paddlepaddle==3.0.0"  # pin 3.0.0 for ARM64
                 if ARM64
-                else "paddlepaddle>=3.0.0"
+                else "paddlepaddle>=3.0.0,!=3.3.0"  # exclude 3.3.0 https://github.com/PaddlePaddle/Paddle/issues/77340
             )
             import paddle.inference as pdi
 
@@ -545,11 +545,17 @@ class AutoBackend(nn.Module):
         # NCNN
         elif ncnn:
             LOGGER.info(f"Loading {w} for NCNN inference...")
+            # use git source for ARM64 due to broken PyPI packages https://github.com/Tencent/ncnn/issues/6509
             check_requirements("git+https://github.com/Tencent/ncnn.git" if ARM64 else "ncnn", cmds="--no-deps")
             import ncnn as pyncnn
 
             net = pyncnn.Net()
-            net.opt.use_vulkan_compute = cuda
+            if isinstance(cuda, torch.device):
+                net.opt.use_vulkan_compute = cuda
+            elif isinstance(device, str) and device.startswith("vulkan"):
+                net.opt.use_vulkan_compute = True
+                net.set_vulkan_device(int(device.split(":")[1]))
+                device = torch.device("cpu")
             w = Path(w)
             if not w.is_file():  # if not *.param
                 w = next(w.glob("*.param"))  # get *.param file from *_ncnn_model dir

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/utils/callbacks/platform.py

@@ -2,6 +2,7 @@
 
 import os
 import platform
+import re
 import socket
 import sys
 from concurrent.futures import ThreadPoolExecutor
@@ -12,6 +13,14 @@ from ultralytics.utils import ENVIRONMENT, GIT, LOGGER, PYTHON_VERSION, RANK, SE
 
 PREFIX = colorstr("Platform: ")
 
+
+def slugify(text):
+    """Convert text to URL-safe slug (e.g., 'My Project 1' -> 'my-project-1')."""
+    if not text:
+        return text
+    return re.sub(r"-+", "-", re.sub(r"[^a-z0-9\s-]", "", str(text).lower()).replace(" ", "-")).strip("-")[:128]
+
+
 try:
     assert not TESTS_RUNNING  # do not log pytest
     assert SETTINGS.get("platform", False) is True or os.getenv("ULTRALYTICS_API_KEY") or SETTINGS.get("api_key")
@@ -249,6 +258,14 @@ def _get_environment_info():
     return env
 
 
+def _get_project_name(trainer):
+    """Get slugified project and name from trainer args."""
+    raw = str(trainer.args.project)
+    parts = raw.split("/", 1)
+    project = f"{parts[0]}/{slugify(parts[1])}" if len(parts) == 2 else slugify(raw)
+    return project, slugify(str(trainer.args.name or "train"))
+
+
 def on_pretrain_routine_start(trainer):
     """Initialize Platform logging at training start."""
     if RANK not in {-1, 0} or not trainer.args.project:
@@ -258,7 +275,7 @@ def on_pretrain_routine_start(trainer):
     trainer._platform_model_id = None
     trainer._platform_last_upload = time()
 
-    project, name = str(trainer.args.project), str(trainer.args.name or "train")
+    project, name = _get_project_name(trainer)
     url = f"https://alpha.ultralytics.com/{project}/{name}"
     LOGGER.info(f"{PREFIX}Streaming to {url}")
 
@@ -305,7 +322,7 @@ def on_fit_epoch_end(trainer):
     if RANK not in {-1, 0} or not trainer.args.project:
         return
 
-    project, name = str(trainer.args.project), str(trainer.args.name or "train")
+    project, name = _get_project_name(trainer)
     metrics = {**trainer.label_loss_items(trainer.tloss, prefix="train"), **trainer.metrics}
 
     if trainer.optimizer and trainer.optimizer.param_groups:
@@ -365,7 +382,7 @@ def on_model_save(trainer):
     if not model_path:
         return
 
-    project, name = str(trainer.args.project), str(trainer.args.name or "train")
+    project, name = _get_project_name(trainer)
     _upload_model_async(model_path, project, name)
     trainer._platform_last_upload = time()
 
@@ -375,7 +392,7 @@ def on_train_end(trainer):
     if RANK not in {-1, 0} or not trainer.args.project:
         return
 
-    project, name = str(trainer.args.project), str(trainer.args.name or "train")
+    project, name = _get_project_name(trainer)
 
     # Stop console capture
     if hasattr(trainer, "_platform_console_logger") and trainer._platform_console_logger:

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/utils/export/imx.py

@@ -219,7 +219,7 @@ def torch2imx(
     Examples:
         >>> from ultralytics import YOLO
         >>> model = YOLO("yolo11n.pt")
-        >>> path, _ = export_imx(model, "model.imx", conf=0.25, iou=0.45, max_det=300)
+        >>> path, _ = export_imx(model, "model.imx", conf=0.25, iou=0.7, max_det=300)
 
     Notes:
         - Requires model_compression_toolkit, onnx, edgemdt_tpc, and edge-mdt-cl packages

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/utils/tal.py

@@ -354,7 +354,8 @@ def make_anchors(feats, strides, grid_cell_offset=0.5):
     anchor_points, stride_tensor = [], []
     assert feats is not None
     dtype, device = feats[0].dtype, feats[0].device
-    for i, stride in enumerate(strides):
+    for i in range(len(feats)):  # use len(feats) to avoid TracerWarning from iterating over strides tensor
+        stride = strides[i]
         h, w = feats[i].shape[2:] if isinstance(feats, list) else (int(feats[i][0]), int(feats[i][1]))
         sx = torch.arange(end=w, device=device, dtype=dtype) + grid_cell_offset  # shift x
         sy = torch.arange(end=h, device=device, dtype=dtype) + grid_cell_offset  # shift y

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/utils/torch_utils.py

@@ -157,7 +157,7 @@ def select_device(device="", newline=False, verbose=True):
     Notes:
         Sets the 'CUDA_VISIBLE_DEVICES' environment variable for specifying which GPUs to use.
     """
-    if isinstance(device, torch.device) or str(device).startswith(("tpu", "intel")):
+    if isinstance(device, torch.device) or str(device).startswith(("tpu", "intel", "vulkan")):
         return device
 
     s = f"Ultralytics {__version__} 🚀 Python-{PYTHON_VERSION} torch-{TORCH_VERSION} "

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253}/ultralytics/utils/tqdm.py

@@ -317,7 +317,10 @@ class TQDM:
             # Final display
             if self.total and self.n >= self.total:
                 self.n = self.total
-            self._display(final=True)
+                if self.n != self.last_print_n:  # Skip if 100% already shown
+                    self._display(final=True)
+            else:
+                self._display(final=True)
 
             # Cleanup
             if self.leave:

{ultralytics_opencv_headless-8.3.251 → ultralytics_opencv_headless-8.3.253/ultralytics_opencv_headless.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-opencv-headless
-Version: 8.3.251
+Version: 8.3.253
 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>