ultralytics 8.0.238__tar.gz → 8.0.239__tar.gz

{ultralytics-8.0.238 → ultralytics-8.0.239}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.0.238
+Version: 8.0.239
 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
@@ -44,6 +44,7 @@ Requires-Dist: py-cpuinfo
 Requires-Dist: thop>=0.1.1
 Requires-Dist: pandas>=1.1.4
 Requires-Dist: seaborn>=0.11.0
+Requires-Dist: hub-sdk>=0.0.2
 Provides-Extra: dev
 Requires-Dist: ipython; extra == "dev"
 Requires-Dist: check-manifest; extra == "dev"

{ultralytics-8.0.238 → ultralytics-8.0.239}/pyproject.toml

@@ -78,6 +78,7 @@ dependencies = [
     "thop>=0.1.1", # FLOPs computation
     "pandas>=1.1.4",
     "seaborn>=0.11.0", # plotting
+    "hub-sdk>=0.0.2", # Ultralytics HUB
 ]
 
 # Optional dependencies ------------------------------------------------------------------------------------------------
@@ -106,9 +107,9 @@ export = [
 ]
 
 explorer = [
-    "lancedb",  # vector search
-    "duckdb",  # SQL queries, supports lancedb tables
-    "streamlit",  # visualizing with GUI
+    "lancedb", # vector search
+    "duckdb", # SQL queries, supports lancedb tables
+    "streamlit", # visualizing with GUI
 ]
 
 # tensorflow>=2.4.1,<=2.13.1  # TF exports (-cpu, -aarch64, -macos)
@@ -178,5 +179,5 @@ pre-summary-newline = true
 close-quotes-on-newline = true
 
 [tool.codespell]
-ignore-words-list = "crate,nd,strack,dota,ane,segway,fo,gool,winn,commend"
-skip = '*.csv,*venv*,docs/??/,docs/mkdocs_??.yml'
+ignore-words-list = "crate,nd,ned,strack,dota,ane,segway,fo,gool,winn,commend,bloc,nam,afterall"
+skip = "*.pt,*.pth,*.torchscript,*.onnx,*.tflite,*.pb,*.bin,*.param,*.mlmodel,*.engine,*.npy,*.data*,*.csv,*pnnx*,*venv*,__pycache__*,*.ico,*.jpg,*.png,*.mp4,*.mov,/runs,/.git,./docs/??/*.md,./docs/mkdocs_??.yml"

{ultralytics-8.0.238 → ultralytics-8.0.239}/tests/test_explorer.py

@@ -1,8 +1,13 @@
+import PIL
+
 from ultralytics import Explorer
 from ultralytics.utils import ASSETS
 
+import PIL
+
 
 def test_similarity():
+    """Test similarity calculations and SQL queries for correctness and response length."""
     exp = Explorer()
     exp.create_embeddings_table()
     similar = exp.get_similar(idx=1)
@@ -18,6 +23,7 @@ def test_similarity():
 
 
 def test_det():
+    """Test detection functionalities and ensure the embedding table has bounding boxes."""
     exp = Explorer(data='coco8.yaml', model='yolov8n.pt')
     exp.create_embeddings_table(force=True)
     assert len(exp.table.head()['bboxes']) > 0
@@ -25,27 +31,26 @@ def test_det():
     assert len(similar) > 0
     # This is a loose test, just checks errors not correctness
     similar = exp.plot_similar(idx=[1, 2], limit=10)
-    assert similar is not None
-    similar.show()
+    assert isinstance(similar, PIL.Image.Image)
 
 
 def test_seg():
+    """Test segmentation functionalities and verify the embedding table includes masks."""
     exp = Explorer(data='coco8-seg.yaml', model='yolov8n-seg.pt')
     exp.create_embeddings_table(force=True)
     assert len(exp.table.head()['masks']) > 0
     similar = exp.get_similar(idx=[1, 2], limit=10)
     assert len(similar) > 0
     similar = exp.plot_similar(idx=[1, 2], limit=10)
-    assert similar is not None
-    similar.show()
+    assert isinstance(similar, PIL.Image.Image)
 
 
 def test_pose():
+    """Test pose estimation functionalities and check the embedding table for keypoints."""
     exp = Explorer(data='coco8-pose.yaml', model='yolov8n-pose.pt')
     exp.create_embeddings_table(force=True)
     assert len(exp.table.head()['keypoints']) > 0
     similar = exp.get_similar(idx=[1, 2], limit=10)
     assert len(similar) > 0
     similar = exp.plot_similar(idx=[1, 2], limit=10)
-    assert similar is not None
-    similar.show()
+    assert isinstance(similar, PIL.Image.Image)

{ultralytics-8.0.238 → ultralytics-8.0.239}/ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = '8.0.238'
+__version__ = "8.0.239"
 
 from ultralytics.data.explorer.explorer import Explorer
 from ultralytics.models import RTDETR, SAM, YOLO
@@ -10,4 +10,4 @@ from ultralytics.utils import SETTINGS as settings
 from ultralytics.utils.checks import check_yolo as checks
 from ultralytics.utils.downloads import download
 
-__all__ = '__version__', 'YOLO', 'NAS', 'SAM', 'FastSAM', 'RTDETR', 'checks', 'download', 'settings', 'Explorer'
+__all__ = "__version__", "YOLO", "NAS", "SAM", "FastSAM", "RTDETR", "checks", "download", "settings", "Explorer"