dgenerate-ultralytics-headless 8.3.184__tar.gz → 8.3.185__tar.gz

{dgenerate_ultralytics_headless-8.3.184 → dgenerate_ultralytics_headless-8.3.185}/PKG-INFO

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dgenerate-ultralytics-headless
-Version: 8.3.184
+Version: 8.3.185
 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.184 → dgenerate_ultralytics_headless-8.3.185}/ultralytics/__init__.py

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

{dgenerate_ultralytics_headless-8.3.184 → dgenerate_ultralytics_headless-8.3.185}/ultralytics/data/converter.py

@@ -613,7 +613,7 @@ def yolo_bbox2segment(
     from ultralytics.utils.ops import xywh2xyxy
 
     # NOTE: add placeholder to pass class index check
-    dataset = YOLODataset(im_dir, data=dict(names=list(range(1000))))
+    dataset = YOLODataset(im_dir, data=dict(names=list(range(1000)), channels=3))
     if len(dataset.labels[0]["segments"]) > 0:  # if it's segment data
         LOGGER.info("Segmentation labels detected, no need to generate new ones!")
         return

{dgenerate_ultralytics_headless-8.3.184 → dgenerate_ultralytics_headless-8.3.185}/ultralytics/data/dataset.py

@@ -85,7 +85,7 @@ class YOLODataset(BaseDataset):
         self.use_obb = task == "obb"
         self.data = data
         assert not (self.use_segments and self.use_keypoints), "Can not use both segments and keypoints."
-        super().__init__(*args, channels=self.data["channels"], **kwargs)
+        super().__init__(*args, channels=self.data.get("channels", 3), **kwargs)
 
     def cache_labels(self, path: Path = Path("./labels.cache")) -> Dict:
         """

{dgenerate_ultralytics_headless-8.3.184 → dgenerate_ultralytics_headless-8.3.185}/ultralytics/utils/__init__.py

@@ -131,8 +131,16 @@ os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"  # suppress "NNPACK.cpp could not in
 os.environ["KINETO_LOG_LEVEL"] = "5"  # suppress verbose PyTorch profiler output when computing FLOPs
 
 if TQDM_RICH := str(os.getenv("YOLO_TQDM_RICH", False)).lower() == "true":
+    from rich.console import Console
+    from rich.progress import BarColumn
     from tqdm import rich
 
+    # Patch Rich Console width=200 and BarColumn bar_width=10 to solve width=80 missing bars bug
+    _console_init = Console.__init__
+    _bar_init = BarColumn.__init__
+    Console.__init__ = lambda self, *a, **k: _console_init(self, *a, **{**k, "width": 200})
+    BarColumn.__init__ = lambda self, bar_width=None, *a, **k: _bar_init(self, 10, *a, **k)
+
 
 class TQDM(rich.tqdm if TQDM_RICH else tqdm.tqdm):
     """

{dgenerate_ultralytics_headless-8.3.184 → dgenerate_ultralytics_headless-8.3.185}/ultralytics/utils/logger.py

@@ -285,16 +285,16 @@ class SystemLogger:
         Collects comprehensive system metrics including CPU usage, RAM usage, disk I/O statistics,
         network I/O statistics, and GPU metrics (if available). Example output:
 
-        ```json
-        {
-            'cpu': 45.2,
-            'ram': 78.9,
-            'disk': {'read_mb': 156.7, 'write_mb': 89.3, 'used_gb': 256.8},
-            'network': {'recv_mb': 157.2, 'sent_mb': 89.1},
-            'gpus': {
-                0: {'usage': 95.6, 'memory': 85.4, 'temp': 72, 'power': 285},
-                1: {'usage': 94.1, 'memory': 82.7, 'temp': 70, 'power': 278}
-            }
+        ```python
+        metrics = {
+            "cpu": 45.2,
+            "ram": 78.9,
+            "disk": {"read_mb": 156.7, "write_mb": 89.3, "used_gb": 256.8},
+            "network": {"recv_mb": 157.2, "sent_mb": 89.1},
+            "gpus": {
+                0: {"usage": 95.6, "memory": 85.4, "temp": 72, "power": 285},
+                1: {"usage": 94.1, "memory": 82.7, "temp": 70, "power": 278},
+            },
         }
         ```
 
@@ -314,7 +314,7 @@ class SystemLogger:
             - power (int): GPU power consumption in watts
 
         Returns:
-            (dict): System metrics containing 'cpu', 'ram', 'disk', 'network', 'gpus' with respective usage data.
+            metrics (dict): System metrics containing 'cpu', 'ram', 'disk', 'network', 'gpus' with respective usage data.
         """
         net = psutil.net_io_counters()
         disk = psutil.disk_io_counters()