ultralytics 8.0.237__tar.gz → 8.0.239__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.0.237
+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"
@@ -266,9 +267,9 @@ See [Pose Docs](https://docs.ultralytics.com/tasks/pose/) for usage examples wit
 
 </details>
 
-<details><summary>Obb (DOTAv1)</summary>
+<details><summary>OBB (DOTAv1)</summary>
 
-See [Obb Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with these models trained on [DOTAv1](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10/), which include 15 pre-trained classes.
+See [OBB Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with these models trained on [DOTAv1](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10/), which include 15 pre-trained classes.
 
 | Model                                                                                        | size<br><sup>(pixels) | mAP<sup>test<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>A100 TensorRT<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
 | -------------------------------------------------------------------------------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
@@ -278,7 +279,7 @@ See [Obb Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with
 | [YOLOv8l-obb](https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l-obb.pt) | 1024                  | 80.7               | 1278.42                        | 11.83                               | 44.5               | 433.8             |
 | [YOLOv8x-obb](https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x-obb.pt) | 1024                  | 81.36              | 1759.10                        | 13.23                               | 69.5               | 676.7             |
 
-- **mAP<sup>test</sup>** values are for single-model multi-scale on [DOTAv1](https://captain-whu.github.io/DOTA/index.html) dataset. <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test`
+- **mAP<sup>test</sup>** values are for single-model multi-scale on [DOTAv1](https://captain-whu.github.io/DOTA/index.html) dataset. <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test` and submit merged results to [DOTA evaluation](https://captain-whu.github.io/DOTA/evaluation.html).
 - **Speed** averaged over DOTAv1 val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. <br>Reproduce by `yolo val obb data=DOTAv1.yaml batch=1 device=0|cpu`
 
 </details>

{ultralytics-8.0.237 → ultralytics-8.0.239}/README.md

@@ -187,9 +187,9 @@ See [Pose Docs](https://docs.ultralytics.com/tasks/pose/) for usage examples wit
 
 </details>
 
-<details><summary>Obb (DOTAv1)</summary>
+<details><summary>OBB (DOTAv1)</summary>
 
-See [Obb Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with these models trained on [DOTAv1](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10/), which include 15 pre-trained classes.
+See [OBB Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with these models trained on [DOTAv1](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10/), which include 15 pre-trained classes.
 
 | Model                                                                                        | size<br><sup>(pixels) | mAP<sup>test<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>A100 TensorRT<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
 | -------------------------------------------------------------------------------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
@@ -199,7 +199,7 @@ See [Obb Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples with
 | [YOLOv8l-obb](https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l-obb.pt) | 1024                  | 80.7               | 1278.42                        | 11.83                               | 44.5               | 433.8             |
 | [YOLOv8x-obb](https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x-obb.pt) | 1024                  | 81.36              | 1759.10                        | 13.23                               | 69.5               | 676.7             |
 
-- **mAP<sup>test</sup>** values are for single-model multi-scale on [DOTAv1](https://captain-whu.github.io/DOTA/index.html) dataset. <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test`
+- **mAP<sup>test</sup>** values are for single-model multi-scale on [DOTAv1](https://captain-whu.github.io/DOTA/index.html) dataset. <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test` and submit merged results to [DOTA evaluation](https://captain-whu.github.io/DOTA/evaluation.html).
 - **Speed** averaged over DOTAv1 val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. <br>Reproduce by `yolo val obb data=DOTAv1.yaml batch=1 device=0|cpu`
 
 </details>

{ultralytics-8.0.237 → 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.237 → ultralytics-8.0.239}/tests/test_cli.py

@@ -13,12 +13,14 @@ TASK_ARGS = [
     ('detect', 'yolov8n', 'coco8.yaml'),
     ('segment', 'yolov8n-seg', 'coco8-seg.yaml'),
     ('classify', 'yolov8n-cls', 'imagenet10'),
-    ('pose', 'yolov8n-pose', 'coco8-pose.yaml'), ]  # (task, model, data)
+    ('pose', 'yolov8n-pose', 'coco8-pose.yaml'),
+    ('obb', 'yolov8n-obb', 'dota8.yaml'), ]  # (task, model, data)
 EXPORT_ARGS = [
     ('yolov8n', 'torchscript'),
     ('yolov8n-seg', 'torchscript'),
     ('yolov8n-cls', 'torchscript'),
-    ('yolov8n-pose', 'torchscript'), ]  # (model, format)
+    ('yolov8n-pose', 'torchscript'),
+    ('yolov8n-obb', 'torchscript'), ]  # (model, format)
 
 
 def run(cmd):

{ultralytics-8.0.237 → 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.237 → ultralytics-8.0.239}/tests/test_python.py

@@ -77,6 +77,7 @@ def test_predict_img():
     seg_model = YOLO(WEIGHTS_DIR / 'yolov8n-seg.pt')
     cls_model = YOLO(WEIGHTS_DIR / 'yolov8n-cls.pt')
     pose_model = YOLO(WEIGHTS_DIR / 'yolov8n-pose.pt')
+    obb_model = YOLO(WEIGHTS_DIR / 'yolov8n-obb.pt')
     im = cv2.imread(str(SOURCE))
     assert len(model(source=Image.open(SOURCE), save=True, verbose=True, imgsz=32)) == 1  # PIL
     assert len(model(source=im, save=True, save_txt=True, imgsz=32)) == 1  # ndarray
@@ -105,6 +106,8 @@ def test_predict_img():
     assert len(results) == t.shape[0]
     results = pose_model(t, imgsz=32)
     assert len(results) == t.shape[0]
+    results = obb_model(t, imgsz=32)
+    assert len(results) == t.shape[0]
 
 
 def test_predict_grey_and_4ch():

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

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = '8.0.237'
+__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"