ultralytics 8.2.50__tar.gz → 8.2.51__tar.gz

{ultralytics-8.2.50/ultralytics.egg-info → ultralytics-8.2.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.2.50
+Version: 8.2.51
 Summary: Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Author: Glenn Jocher, Ayush Chaurasia, Jing Qiu
 Maintainer: Glenn Jocher, Ayush Chaurasia, Jing Qiu

{ultralytics-8.2.50 → ultralytics-8.2.51}/ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = "8.2.50"
+__version__ = "8.2.51"
 
 import os
 

{ultralytics-8.2.50 → ultralytics-8.2.51}/ultralytics/engine/exporter.py

@@ -686,9 +686,10 @@ class Exporter:
             import tensorrt as trt  # noqa
         except ImportError:
             if LINUX:
-                check_requirements("tensorrt", cmds="-U")
+                check_requirements("tensorrt>7.0.0,<=10.1.0")
             import tensorrt as trt  # noqa
-        check_version(trt.__version__, "7.0.0", hard=True)  # require tensorrt>=7.0.0
+        check_version(trt.__version__, ">=7.0.0", hard=True)
+        check_version(trt.__version__, "<=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
 
         # Setup and checks
         LOGGER.info(f"\n{prefix} starting export with TensorRT {trt.__version__}...")

{ultralytics-8.2.50 → ultralytics-8.2.51}/ultralytics/engine/results.py

@@ -120,7 +120,12 @@ class Results(SimpleClass):
             boxes (torch.tensor, optional): A 2D tensor of bounding box coordinates for each detection.
             masks (torch.tensor, optional): A 3D tensor of detection masks, where each mask is a binary image.
             probs (torch.tensor, optional): A 1D tensor of probabilities of each class for classification task.
-            keypoints (torch.tensor, optional): A 2D tensor of keypoint coordinates for each detection.
+            keypoints (torch.tensor, optional): A 2D tensor of keypoint coordinates for each detection. For default pose
+                model, Keypoint indices for human body pose estimation are:
+                0: Nose, 1: Left Eye, 2: Right Eye, 3: Left Ear, 4: Right Ear
+                5: Left Shoulder, 6: Right Shoulder, 7: Left Elbow, 8: Right Elbow
+                9: Left Wrist, 10: Right Wrist, 11: Left Hip, 12: Right Hip
+                13: Left Knee, 14: Right Knee, 15: Left Ankle, 16: Right Ankle
             obb (torch.tensor, optional): A 2D tensor of oriented bounding box coordinates for each detection.
             speed (dict, optional): A dictionary containing preprocess, inference, and postprocess speeds (ms/image).
 

{ultralytics-8.2.50 → ultralytics-8.2.51}/ultralytics/nn/autobackend.py

@@ -225,9 +225,10 @@ class AutoBackend(nn.Module):
                 import tensorrt as trt  # noqa https://developer.nvidia.com/nvidia-tensorrt-download
             except ImportError:
                 if LINUX:
-                    check_requirements("tensorrt", cmds="-U")
+                    check_requirements("tensorrt>7.0.0,<=10.1.0")
                 import tensorrt as trt  # noqa
-            check_version(trt.__version__, "7.0.0", hard=True)  # require tensorrt>=7.0.0
+            check_version(trt.__version__, ">=7.0.0", hard=True)
+            check_version(trt.__version__, "<=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
             if device.type == "cpu":
                 device = torch.device("cuda:0")
             Binding = namedtuple("Binding", ("name", "dtype", "shape", "data", "ptr"))

{ultralytics-8.2.50 → ultralytics-8.2.51}/ultralytics/utils/downloads.py

@@ -22,7 +22,7 @@ GITHUB_ASSETS_NAMES = (
     + [f"yolov3{k}u.pt" for k in ("", "-spp", "-tiny")]
     + [f"yolov8{k}-world.pt" for k in "smlx"]
     + [f"yolov8{k}-worldv2.pt" for k in "smlx"]
-    + [f"yolov9{k}.pt" for k in "ce"]
+    + [f"yolov9{k}.pt" for k in "tsmce"]
     + [f"yolov10{k}.pt" for k in "nsmblx"]
     + [f"yolo_nas_{k}.pt" for k in "sml"]
     + [f"sam_{k}.pt" for k in "bl"]

{ultralytics-8.2.50 → ultralytics-8.2.51/ultralytics.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.2.50
+Version: 8.2.51
 Summary: Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Author: Glenn Jocher, Ayush Chaurasia, Jing Qiu
 Maintainer: Glenn Jocher, Ayush Chaurasia, Jing Qiu