ultralytics 8.2.54__tar.gz → 8.2.56__tar.gz

{ultralytics-8.2.54/ultralytics.egg-info → ultralytics-8.2.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.2.54
+Version: 8.2.56
 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.54 → ultralytics-8.2.56}/ultralytics/__init__.py

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

{ultralytics-8.2.54 → ultralytics-8.2.56}/ultralytics/engine/exporter.py

@@ -1017,13 +1017,13 @@ class Exporter:
         """Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata."""
         import flatbuffers
 
-        if ARM64:
-            from tflite_support import metadata  # noqa
-            from tflite_support import metadata_schema_py_generated as schema  # noqa
-        else:
+        try:
             # TFLite Support bug https://github.com/tensorflow/tflite-support/issues/954#issuecomment-2108570845
             from tensorflow_lite_support.metadata import metadata_schema_py_generated as schema  # noqa
             from tensorflow_lite_support.metadata.python import metadata  # noqa
+        except ImportError:  # ARM64 systems may not have the 'tensorflow_lite_support' package available
+            from tflite_support import metadata  # noqa
+            from tflite_support import metadata_schema_py_generated as schema  # noqa
 
         # Create model info
         model_meta = schema.ModelMetadataT()

{ultralytics-8.2.54 → ultralytics-8.2.56}/ultralytics/hub/utils.py

@@ -185,7 +185,7 @@ class Events:
     def __init__(self):
         """Initializes the Events object with default values for events, rate_limit, and metadata."""
         self.events = []  # events list
-        self.rate_limit = 60.0  # rate limit (seconds)
+        self.rate_limit = 30.0  # rate limit (seconds)
         self.t = 0.0  # rate limit timer (seconds)
         self.metadata = {
             "cli": Path(ARGV[0]).name == "yolo",

{ultralytics-8.2.54 → ultralytics-8.2.56}/ultralytics/solutions/streamlit_inference.py

@@ -6,6 +6,8 @@ import time
 import cv2
 import torch
 
+from ultralytics.utils.downloads import GITHUB_ASSETS_STEMS
+
 
 def inference():
     """Runs real-time object detection on video input using Ultralytics YOLOv8 in a Streamlit application."""
@@ -65,28 +67,12 @@ def inference():
         vid_file_name = 0
 
     # Add dropdown menu for model selection
-    yolov8_model = st.sidebar.selectbox(
-        "Model",
-        (
-            "YOLOv8n",
-            "YOLOv8s",
-            "YOLOv8m",
-            "YOLOv8l",
-            "YOLOv8x",
-            "YOLOv8n-Seg",
-            "YOLOv8s-Seg",
-            "YOLOv8m-Seg",
-            "YOLOv8l-Seg",
-            "YOLOv8x-Seg",
-            "YOLOv8n-Pose",
-            "YOLOv8s-Pose",
-            "YOLOv8m-Pose",
-            "YOLOv8l-Pose",
-            "YOLOv8x-Pose",
-        ),
-    )
-    model = YOLO(f"{yolov8_model.lower()}.pt")  # Load the yolov8 model
-    class_names = list(model.names.values())  # Convert dictionary to list of class names
+    available_models = (x.replace("yolo", "YOLO") for x in GITHUB_ASSETS_STEMS if x.startswith("yolov8"))
+    selected_model = st.sidebar.selectbox("Model", available_models)
+    with st.spinner("Model is downloading..."):
+        model = YOLO(f"{selected_model.lower()}.pt")  # Load the YOLO model
+        class_names = list(model.names.values())  # Convert dictionary to list of class names
+    st.success("Model loaded successfully!")
 
     # Multiselect box with class names and get indices of selected classes
     selected_classes = st.sidebar.multiselect("Classes", class_names, default=class_names[:3])
@@ -95,8 +81,9 @@ def inference():
     if not isinstance(selected_ind, list):  # Ensure selected_options is a list
         selected_ind = list(selected_ind)
 
-    conf_thres = st.sidebar.slider("Confidence Threshold", 0.0, 1.0, 0.25, 0.01)
-    nms_thres = st.sidebar.slider("NMS Threshold", 0.0, 1.0, 0.45, 0.01)
+    enable_trk = st.sidebar.radio("Enable Tracking", ("Yes", "No"))
+    conf = float(st.sidebar.slider("Confidence Threshold", 0.0, 1.0, 0.25, 0.01))
+    iou = float(st.sidebar.slider("IoU Threshold", 0.0, 1.0, 0.45, 0.01))
 
     col1, col2 = st.columns(2)
     org_frame = col1.empty()
@@ -124,7 +111,10 @@ def inference():
             prev_time = curr_time
 
             # Store model predictions
-            results = model(frame, conf=float(conf_thres), iou=float(nms_thres), classes=selected_ind)
+            if enable_trk:
+                results = model.track(frame, conf=conf, iou=iou, classes=selected_ind, persist=True)
+            else:
+                results = model(frame, conf=conf, iou=iou, classes=selected_ind)
             annotated_frame = results[0].plot()  # Add annotations on frame
 
             # display frame

{ultralytics-8.2.54 → ultralytics-8.2.56}/ultralytics/utils/torch_utils.py

@@ -271,7 +271,7 @@ def model_info(model, detailed=False, verbose=True, imgsz=640):
     fs = f", {flops:.1f} GFLOPs" if flops else ""
     yaml_file = getattr(model, "yaml_file", "") or getattr(model, "yaml", {}).get("yaml_file", "")
     model_name = Path(yaml_file).stem.replace("yolo", "YOLO") or "Model"
-    LOGGER.info(f"{model_name} summary{fused}: {n_l} layers, {n_p} parameters, {n_g} gradients{fs}")
+    LOGGER.info(f"{model_name} summary{fused}: {n_l:,} layers, {n_p:,} parameters, {n_g:,} gradients{fs}")
     return n_l, n_p, n_g, flops
 
 
@@ -513,6 +513,9 @@ def strip_optimizer(f: Union[str, Path] = "best.pt", s: str = "") -> None:
         for f in Path('path/to/model/checkpoints').rglob('*.pt'):
             strip_optimizer(f)
         ```
+
+    Note:
+        Use `ultralytics.nn.torch_safe_load` for missing modules with `x = torch_safe_load(f)[0]`
     """
     try:
         x = torch.load(f, map_location=torch.device("cpu"))

{ultralytics-8.2.54 → ultralytics-8.2.56/ultralytics.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.2.54
+Version: 8.2.56
 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