ultralytics 8.2.72__py3-none-any.whl → 8.2.74__py3-none-any.whl

ultralytics/trackers/byte_tracker.py

@@ -375,9 +375,8 @@ class BYTETracker:
     def get_dists(self, tracks, detections):
         """Calculates the distance between tracks and detections using IoU and fuses scores."""
         dists = matching.iou_distance(tracks, detections)
-        # TODO: mot20
-        # if not self.args.mot20:
-        dists = matching.fuse_score(dists, detections)
+        if self.args.fuse_score:
+            dists = matching.fuse_score(dists, detections)
         return dists
 
     def multi_predict(self, tracks):

{ultralytics-8.2.72.dist-info → ultralytics-8.2.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.2.72
+Version: 8.2.74
 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
@@ -30,56 +30,56 @@ Classifier: Operating System :: Microsoft :: Windows
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: numpy <2.0.0,>=1.23.0
-Requires-Dist: matplotlib >=3.3.0
-Requires-Dist: opencv-python >=4.6.0
-Requires-Dist: pillow >=7.1.2
-Requires-Dist: pyyaml >=5.3.1
-Requires-Dist: requests >=2.23.0
-Requires-Dist: scipy >=1.4.1
-Requires-Dist: torch >=1.8.0
-Requires-Dist: torchvision >=0.9.0
-Requires-Dist: tqdm >=4.64.0
+Requires-Dist: numpy<2.0.0,>=1.23.0
+Requires-Dist: matplotlib>=3.3.0
+Requires-Dist: opencv-python>=4.6.0
+Requires-Dist: pillow>=7.1.2
+Requires-Dist: pyyaml>=5.3.1
+Requires-Dist: requests>=2.23.0
+Requires-Dist: scipy>=1.4.1
+Requires-Dist: torch>=1.8.0
+Requires-Dist: torchvision>=0.9.0
+Requires-Dist: tqdm>=4.64.0
 Requires-Dist: psutil
 Requires-Dist: py-cpuinfo
-Requires-Dist: pandas >=1.1.4
-Requires-Dist: seaborn >=0.11.0
-Requires-Dist: ultralytics-thop >=2.0.0
+Requires-Dist: pandas>=1.1.4
+Requires-Dist: seaborn>=0.11.0
+Requires-Dist: ultralytics-thop>=2.0.0
 Provides-Extra: dev
-Requires-Dist: ipython ; extra == 'dev'
-Requires-Dist: pytest ; extra == 'dev'
-Requires-Dist: pytest-cov ; extra == 'dev'
-Requires-Dist: coverage[toml] ; extra == 'dev'
-Requires-Dist: mkdocs >=1.6.0 ; extra == 'dev'
-Requires-Dist: mkdocs-material >=9.5.9 ; extra == 'dev'
-Requires-Dist: mkdocstrings[python] ; extra == 'dev'
-Requires-Dist: mkdocs-jupyter ; extra == 'dev'
-Requires-Dist: mkdocs-redirects ; extra == 'dev'
-Requires-Dist: mkdocs-ultralytics-plugin >=0.0.49 ; extra == 'dev'
+Requires-Dist: ipython; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: coverage[toml]; extra == "dev"
+Requires-Dist: mkdocs>=1.6.0; extra == "dev"
+Requires-Dist: mkdocs-material>=9.5.9; extra == "dev"
+Requires-Dist: mkdocstrings[python]; extra == "dev"
+Requires-Dist: mkdocs-jupyter; extra == "dev"
+Requires-Dist: mkdocs-redirects; extra == "dev"
+Requires-Dist: mkdocs-ultralytics-plugin>=0.0.49; extra == "dev"
 Provides-Extra: explorer
-Requires-Dist: lancedb ; extra == 'explorer'
-Requires-Dist: duckdb <=0.9.2 ; extra == 'explorer'
-Requires-Dist: streamlit ; extra == 'explorer'
+Requires-Dist: lancedb; extra == "explorer"
+Requires-Dist: duckdb<=0.9.2; extra == "explorer"
+Requires-Dist: streamlit; extra == "explorer"
 Provides-Extra: export
-Requires-Dist: onnx >=1.12.0 ; extra == 'export'
-Requires-Dist: openvino >=2024.0.0 ; extra == 'export'
-Requires-Dist: tensorflow >=2.0.0 ; extra == 'export'
-Requires-Dist: tensorflowjs >=3.9.0 ; extra == 'export'
-Requires-Dist: keras ; extra == 'export'
-Requires-Dist: flatbuffers <100,>=23.5.26 ; (platform_machine == "aarch64") and extra == 'export'
-Requires-Dist: numpy ==1.23.5 ; (platform_machine == "aarch64") and extra == 'export'
-Requires-Dist: h5py !=3.11.0 ; (platform_machine == "aarch64") and extra == 'export'
-Requires-Dist: tensorstore >=0.1.63 ; (platform_machine == "aarch64" and python_version >= "3.9") and extra == 'export'
-Requires-Dist: coremltools >=7.0 ; (platform_system != "Windows" and python_version <= "3.11") and extra == 'export'
+Requires-Dist: onnx>=1.12.0; extra == "export"
+Requires-Dist: openvino>=2024.0.0; extra == "export"
+Requires-Dist: tensorflow>=2.0.0; extra == "export"
+Requires-Dist: tensorflowjs>=3.9.0; extra == "export"
+Requires-Dist: keras; extra == "export"
+Requires-Dist: flatbuffers<100,>=23.5.26; platform_machine == "aarch64" and extra == "export"
+Requires-Dist: numpy==1.23.5; platform_machine == "aarch64" and extra == "export"
+Requires-Dist: h5py!=3.11.0; platform_machine == "aarch64" and extra == "export"
+Requires-Dist: tensorstore>=0.1.63; (platform_machine == "aarch64" and python_version >= "3.9") and extra == "export"
+Requires-Dist: coremltools>=7.0; (platform_system != "Windows" and python_version <= "3.11") and extra == "export"
 Provides-Extra: extra
-Requires-Dist: hub-sdk >=0.0.8 ; extra == 'extra'
-Requires-Dist: ipython ; extra == 'extra'
-Requires-Dist: albumentations >=1.4.6 ; extra == 'extra'
-Requires-Dist: pycocotools >=2.0.7 ; extra == 'extra'
+Requires-Dist: hub-sdk>=0.0.8; extra == "extra"
+Requires-Dist: ipython; extra == "extra"
+Requires-Dist: albumentations>=1.4.6; extra == "extra"
+Requires-Dist: pycocotools>=2.0.7; extra == "extra"
 Provides-Extra: logging
-Requires-Dist: comet ; extra == 'logging'
-Requires-Dist: tensorboard >=2.13.0 ; extra == 'logging'
-Requires-Dist: dvclive >=2.12.0 ; extra == 'logging'
+Requires-Dist: comet; extra == "logging"
+Requires-Dist: tensorboard>=2.13.0; extra == "logging"
+Requires-Dist: dvclive>=2.12.0; extra == "logging"
 
 <div align="center">
   <p>

{ultralytics-8.2.72.dist-info → ultralytics-8.2.74.dist-info}/RECORD

@@ -8,7 +8,7 @@ tests/test_exports.py,sha256=Uezf3OatpPHlo5qoPw-2kqkZxuMCF9L4XF2riD4vmII,8225
 tests/test_integrations.py,sha256=xglcfMPjfVh346PV8WTpk6tBxraCXEFJEQyyJMr5tyU,6064
 tests/test_python.py,sha256=cLK8dyRf_4H_znFIm-krnOFMydwkxKlVZvHwl9vbck8,21780
 tests/test_solutions.py,sha256=EACnPXbeJe2aVTOKfqMk5jclKKCWCVgFEzjpR6y7Sh8,3304
-ultralytics/__init__.py,sha256=kF5QJe8JpLVYC89NNXSzMiIAAV8VUEy3d2bWGR6OQmA,712
+ultralytics/__init__.py,sha256=0mePJ1OrsmtoXd_P7bhGJ4w9rk7MWbBmC-mt2qqq-zM,694
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
 ultralytics/cfg/__init__.py,sha256=7ce3_bhi7pDw5ZAbSqYR6e3_IYD2JCLCy7fkl5d1WyI,33064
@@ -80,8 +80,8 @@ ultralytics/cfg/models/v9/yolov9e.yaml,sha256=dhaR47WxuLOrZWDCceS4bQG00sQdrMc8FQ
 ultralytics/cfg/models/v9/yolov9m.yaml,sha256=l6CmivzNu44sRVmkQXk4-tXflbV1nWnk5MSc8su2vhs,1311
 ultralytics/cfg/models/v9/yolov9s.yaml,sha256=lPWcu-6ub1kCBD6zIDFwthYZ3RvdJfODWKy3vEQWRjo,1291
 ultralytics/cfg/models/v9/yolov9t.yaml,sha256=qL__kr6GoefpQWP4jV0jdzwTp46bdFUcqtPRnfDbkY8,1275
-ultralytics/cfg/trackers/botsort.yaml,sha256=YrPmj18p1UU40kJH5NRdL_4S8f7knggkk_q2KYnVudo,883
-ultralytics/cfg/trackers/bytetrack.yaml,sha256=QvHmtuwulK4X6j3T5VEqtCm0sbWWBUVmWPcCcM20qe0,688
+ultralytics/cfg/trackers/botsort.yaml,sha256=8B0xNbnG_E-9DCUpap72PWkUgBb1AjuApEn7gHiVngE,916
+ultralytics/cfg/trackers/bytetrack.yaml,sha256=8vpTZ2x9mhRXJymoJvs1G8kTXo_HxbSwHup2FQALT3A,721
 ultralytics/data/__init__.py,sha256=VGe-ATG7j35F4A4r8Jmzffjlhve4JAJPgRa5ahKTU18,616
 ultralytics/data/annotator.py,sha256=1Hyu6ubrBL8KmRrt1keGn-K4XTqQdAVyIwTsQiBtzLU,2489
 ultralytics/data/augment.py,sha256=ExU4khJfJ_TeczkJRLNUDscN57SJvAjnm-reouJcxGI,119309
@@ -110,7 +110,7 @@ ultralytics/hub/auth.py,sha256=FID58NE6fh7Op_B45QOpWBw1qoBN0ponL16uvyb2dZ8,5399
 ultralytics/hub/session.py,sha256=UF_aVwyxnbP-OzpzKXGGhi4i6KGWjjhoj5Qsn46dFpE,16257
 ultralytics/hub/utils.py,sha256=tXfM3QbXBcf4Y6StgHI1pktT4OM7Ic9eF3xiBFHGlhY,9721
 ultralytics/hub/google/__init__.py,sha256=qyvvpGP-4NAtrn7GLqfqxP_aWuRP1T0OvJYafWKvL2Q,7512
-ultralytics/models/__init__.py,sha256=AlVStwxv5pMrYaPL8dLhu4sY2c2JgqmuK__RlEDKrEo,296
+ultralytics/models/__init__.py,sha256=TT9iLCL_n9Y80dcUq0Fo-p-GRZCSU2vrWXM3CoMwqqE,265
 ultralytics/models/fastsam/__init__.py,sha256=W0rRSJM3vdxcsneuiN6_ajkUw86k6-opUKdLxVhKOoQ,203
 ultralytics/models/fastsam/model.py,sha256=r5VZj-KLKaqZtEKTZxQik8vQI2N9uOF4xpV_gA-P8h0,2101
 ultralytics/models/fastsam/predict.py,sha256=z5j2IMwf4MURuROKeqNXW1WvOSj91UdJa7dLRqN_OFc,7370
@@ -125,28 +125,20 @@ ultralytics/models/rtdetr/model.py,sha256=2VkppF1_581XmQ0UI7lo8fX7MqhAJPXVMr2jyM
 ultralytics/models/rtdetr/predict.py,sha256=GmeNiFszDajq9YNPi0jW89CqP0MRD5Gtmokh9z0JAQc,3568
 ultralytics/models/rtdetr/train.py,sha256=20AFYVW9NPxw0-cp-sRdIovWidFL0IIhJRv2oZjkPlM,3685
 ultralytics/models/rtdetr/val.py,sha256=4QQArdaGEY8rJsJuvyJ032f8GGVGdV2jURHK2EdMxyk,5566
-ultralytics/models/sam/__init__.py,sha256=9A1iyfPN_ncqq3TMExe_-uPoARjEX3psoHEI1xMG2VE,144
-ultralytics/models/sam/amg.py,sha256=He2c4nIoZ__F_pL18rRl278R8iBjWXBM2Z_vxfuVOkk,7971
-ultralytics/models/sam/build.py,sha256=BSpRgDIQb-kgxaQtSq0C7Zb2UsqkfkFRmErC_bzKYIg,4954
-ultralytics/models/sam/model.py,sha256=MVO7WqF41Sq1-qbsN8O8Fophe8anYVY67yp17Sudp0k,4979
-ultralytics/models/sam/predict.py,sha256=GoR8xCwt3VJdBX5wUUhJ-3qY87LoG-R1eG8SziTDLP0,23755
+ultralytics/models/sam/__init__.py,sha256=o4_D6y8YJlOXIK7Lwo9RHnIJJ9xoFNi4zK99QSc1kdM,176
+ultralytics/models/sam/amg.py,sha256=GrmO_8YfIDt_QkPEMF_WFjPZkhwhf7iwx7ig8JgOUnE,8709
+ultralytics/models/sam/build.py,sha256=zNQbrgSHUgz1gyXQwLKGTpa6CSEjeaevcP3w1Z1l3mo,12233
+ultralytics/models/sam/model.py,sha256=uOm5xEPX9PJYu6ag5T5iHzl4Uq8huwPyoaYo9imf9HQ,7374
+ultralytics/models/sam/predict.py,sha256=ILPx2O4pj4hmMG1KGwHjMYBM3pD-4mZLPNqIczk6_S0,37744
 ultralytics/models/sam/modules/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
-ultralytics/models/sam/modules/decoders.py,sha256=SkVdfsFwy8g4rOJYXi2rWg2zI5HEttsQJi6E4Uwxs9o,6307
-ultralytics/models/sam/modules/encoders.py,sha256=JHot5dRyr_d3wAzz7jgc_SOWOWibWKRWOt-IGBXPhW8,24894
-ultralytics/models/sam/modules/sam.py,sha256=lyB-edOBr85gACTaVqG0WiSIS4FyohTtLqkNMKDwVM0,2695
-ultralytics/models/sam/modules/tiny_encoder.py,sha256=rAY9JuyxUpFivFUUPVjK2aUYlsXEZ0JGKVoEWDGf0Eo,29228
-ultralytics/models/sam/modules/transformer.py,sha256=a2jsS_J76MvrIKIERb_0flliYFMjpBbwVL4UnsNnoyE,11232
-ultralytics/models/sam2/__init__.py,sha256=_xqQHLZTLgEdK278YETYR-Fts2hsvXP5q9ddUbuuFvc,154
-ultralytics/models/sam2/build.py,sha256=m6hv82VKn3Lct_7nztUqdzJzCV9Nbr5mvqpI8nkReQM,5422
-ultralytics/models/sam2/model.py,sha256=PS-eV78DVNrGZmUq7L7gJHgrGjxnySM1TTHkwfrQM7E,3408
-ultralytics/models/sam2/predict.py,sha256=I_ZM3oA2-6Y2gjWGJWsDmQeLM51JSVRBZNGzNwRszY4,8636
-ultralytics/models/sam2/modules/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
-ultralytics/models/sam2/modules/decoders.py,sha256=t4SR-0g3HQstk-agiapCsVYTMZBFc2vz24zfgBwZUkw,15376
-ultralytics/models/sam2/modules/encoders.py,sha256=0VRK2wdl0vZzKA3528_j-Vyn4Iy8XlNHp2ftQRn-aGE,13313
-ultralytics/models/sam2/modules/memory_attention.py,sha256=4zdvm8_ANM0r8QSN_xBGi9l-9Ugjt3gxBsHv2cHczjc,6214
-ultralytics/models/sam2/modules/sam2.py,sha256=CgCBrfjhKDHI2n8iM6AIJmXeCEgf2_qUz7rzZT31fB0,44255
-ultralytics/models/sam2/modules/sam2_blocks.py,sha256=7HmuZTFw8VVdAVDsIStWByxyUHBqytnfgvQMaCNr1GU,28379
-ultralytics/models/sam2/modules/utils.py,sha256=2H5C3sjBnYoPuoJqflH3AmGeBJoKrhHea136jgwIq_I,8320
+ultralytics/models/sam/modules/blocks.py,sha256=qXCXMqkQG0fpAvCkA9TrtimfOLDtyJfCx3bDfh3bJUs,45974
+ultralytics/models/sam/modules/decoders.py,sha256=d02t-55eTUBXEUtaDbcLm4VLgfqsW478CpJouMkMK-g,25874
+ultralytics/models/sam/modules/encoders.py,sha256=KvQFAtqfGvCAr4kcMXxnJvjwIhaQ0a3Wwp0KhSSG_oA,34615
+ultralytics/models/sam/modules/memory_attention.py,sha256=XilWBnRfH8wZxIoL2-yEk-dRypCsS0Jf_9t8WJxXKg0,9722
+ultralytics/models/sam/modules/sam.py,sha256=1U2XGskHymvhZ8OFKYzTEfAI9zJuqWn-d1qonC57dfQ,49397
+ultralytics/models/sam/modules/tiny_encoder.py,sha256=yFbrPMuFuA9LIMFVhA2VvkhwE-JOKdrg5euTkgilPIk,40437
+ultralytics/models/sam/modules/transformer.py,sha256=cUxCPOBfsl3GNhuhC1b9l4j5FuJTb3_e7jD4td5TFhU,16156
+ultralytics/models/sam/modules/utils.py,sha256=HVQiV5P_a1JH_9hZ71Cc0yAkCjw2iML20b_CPErahlE,12315
 ultralytics/models/utils/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
 ultralytics/models/utils/loss.py,sha256=PmlKDe4xQTiYkPSCdNUabxJC7bh43zGxiKVIxsXBVGE,15135
 ultralytics/models/utils/ops.py,sha256=sAeD_koytXDzHibIvQLLAx3vOpGdhdAiQhMiNFUnn5U,13255
@@ -197,8 +189,8 @@ ultralytics/solutions/speed_estimation.py,sha256=kjqMSHGTHMZaNgTKNKWULxnJQNsvhq4
 ultralytics/solutions/streamlit_inference.py,sha256=MKf5P3O5oJwIKu2h_URvzaQjMWoSEMDMBwordplfRxo,5703
 ultralytics/trackers/__init__.py,sha256=j72IgH2dZHQArMPK4YwcV5ieIw94fYvlGdQjB9cOQKw,227
 ultralytics/trackers/basetrack.py,sha256=-vBDD-Q9lsxfTMK2w9kuqWGrYbRMmaBCCEbGGyR53gE,3675
-ultralytics/trackers/bot_sort.py,sha256=39AvhYVbT7izF3--rX_e6Lhgb5czTA23gw6AgnNcRds,8601
-ultralytics/trackers/byte_tracker.py,sha256=X3IAl9v6GyCv5vLPBvWyUpzpLHk7Sq9CPpYKPsVlg2M,18857
+ultralytics/trackers/bot_sort.py,sha256=7wuuQpsMSfJ9m-fzkxpx0mDjh_7Wkhv9lheNpfsuyDs,8582
+ultralytics/trackers/byte_tracker.py,sha256=0KPnGAKyCXGHb6G7EPhio8mFdJTiB_WC2oqTI0pys6U,18838
 ultralytics/trackers/track.py,sha256=NnCA99jXVOvj7XpL_RvMm01DGdixKqAMFz6Sdds30kQ,3436
 ultralytics/trackers/utils/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
 ultralytics/trackers/utils/gmc.py,sha256=-1oBNFRB-9EawJmUOT566AygLCVxJw-jsPSIOl5j_Hk,13683
@@ -233,9 +225,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=5Z3ua5YBTUS56FH8VQKQG1aaIo9fH8GEyz
 ultralytics/utils/callbacks/raytune.py,sha256=ODVYzy-CoM4Uge0zjkh3Hnh9nF2M0vhDrSenXnvcizw,705
 ultralytics/utils/callbacks/tensorboard.py,sha256=QEgOVhUqY9akOs5TJIwz1Rvn6l32xWLpOxlwEyWF0B8,4136
 ultralytics/utils/callbacks/wb.py,sha256=9-fjQIdLjr3b73DTE3rHO171KvbH1VweJ-bmbv-rqTw,6747
-ultralytics-8.2.72.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
-ultralytics-8.2.72.dist-info/METADATA,sha256=LGYZMqpyQuiT5ZBm8veVX1TFybGLIXFSDHXKzOjAfGw,41337
-ultralytics-8.2.72.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-ultralytics-8.2.72.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
-ultralytics-8.2.72.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-ultralytics-8.2.72.dist-info/RECORD,,
+ultralytics-8.2.74.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
+ultralytics-8.2.74.dist-info/METADATA,sha256=b5wD8pNfYLoERi_TndgKk2c76llIAvwz2eB47WP9WKA,41270
+ultralytics-8.2.74.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+ultralytics-8.2.74.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
+ultralytics-8.2.74.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+ultralytics-8.2.74.dist-info/RECORD,,

ultralytics/models/sam2/__init__.py

@@ -1,6 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
-
-from .model import SAM2
-from .predict import SAM2Predictor
-
-__all__ = "SAM2", "SAM2Predictor"  # tuple or list

ultralytics/models/sam2/build.py

@@ -1,156 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
-
-import torch
-
-from ultralytics.utils.downloads import attempt_download_asset
-
-from .modules.encoders import FpnNeck, Hiera, ImageEncoder, MemoryEncoder
-from .modules.memory_attention import MemoryAttention, MemoryAttentionLayer
-from .modules.sam2 import SAM2Model
-
-
-def build_sam2_t(checkpoint=None):
-    """Build and return a Segment Anything Model (SAM2) tiny-size model with specified architecture parameters."""
-    return _build_sam2(
-        encoder_embed_dim=96,
-        encoder_stages=[1, 2, 7, 2],
-        encoder_num_heads=1,
-        encoder_global_att_blocks=[5, 7, 9],
-        encoder_window_spec=[8, 4, 14, 7],
-        encoder_backbone_channel_list=[768, 384, 192, 96],
-        checkpoint=checkpoint,
-    )
-
-
-def build_sam2_s(checkpoint=None):
-    """Builds and returns a small-size Segment Anything Model (SAM2) with specified architecture parameters."""
-    return _build_sam2(
-        encoder_embed_dim=96,
-        encoder_stages=[1, 2, 11, 2],
-        encoder_num_heads=1,
-        encoder_global_att_blocks=[7, 10, 13],
-        encoder_window_spec=[8, 4, 14, 7],
-        encoder_backbone_channel_list=[768, 384, 192, 96],
-        checkpoint=checkpoint,
-    )
-
-
-def build_sam2_b(checkpoint=None):
-    """Builds and returns a Segment Anything Model (SAM2) base-size model with specified architecture parameters."""
-    return _build_sam2(
-        encoder_embed_dim=112,
-        encoder_stages=[2, 3, 16, 3],
-        encoder_num_heads=2,
-        encoder_global_att_blocks=[12, 16, 20],
-        encoder_window_spec=[8, 4, 14, 7],
-        encoder_window_spatial_size=[14, 14],
-        encoder_backbone_channel_list=[896, 448, 224, 112],
-        checkpoint=checkpoint,
-    )
-
-
-def build_sam2_l(checkpoint=None):
-    """Build and return a Segment Anything Model (SAM2) large-size model with specified architecture parameters."""
-    return _build_sam2(
-        encoder_embed_dim=144,
-        encoder_stages=[2, 6, 36, 4],
-        encoder_num_heads=2,
-        encoder_global_att_blocks=[23, 33, 43],
-        encoder_window_spec=[8, 4, 16, 8],
-        encoder_backbone_channel_list=[1152, 576, 288, 144],
-        checkpoint=checkpoint,
-    )
-
-
-def _build_sam2(
-    encoder_embed_dim=1280,
-    encoder_stages=[2, 6, 36, 4],
-    encoder_num_heads=2,
-    encoder_global_att_blocks=[7, 15, 23, 31],
-    encoder_backbone_channel_list=[1152, 576, 288, 144],
-    encoder_window_spatial_size=[7, 7],
-    encoder_window_spec=[8, 4, 16, 8],
-    checkpoint=None,
-):
-    """Builds a SAM2 model with specified architecture parameters and optional checkpoint loading."""
-    image_encoder = ImageEncoder(
-        trunk=Hiera(
-            embed_dim=encoder_embed_dim,
-            num_heads=encoder_num_heads,
-            stages=encoder_stages,
-            global_att_blocks=encoder_global_att_blocks,
-            window_pos_embed_bkg_spatial_size=encoder_window_spatial_size,
-            window_spec=encoder_window_spec,
-        ),
-        neck=FpnNeck(
-            d_model=256,
-            backbone_channel_list=encoder_backbone_channel_list,
-            fpn_top_down_levels=[2, 3],
-            fpn_interp_model="nearest",
-        ),
-        scalp=1,
-    )
-    memory_attention = MemoryAttention(d_model=256, pos_enc_at_input=True, num_layers=4, layer=MemoryAttentionLayer())
-    memory_encoder = MemoryEncoder(out_dim=64)
-
-    sam2 = SAM2Model(
-        image_encoder=image_encoder,
-        memory_attention=memory_attention,
-        memory_encoder=memory_encoder,
-        num_maskmem=7,
-        image_size=1024,
-        sigmoid_scale_for_mem_enc=20.0,
-        sigmoid_bias_for_mem_enc=-10.0,
-        use_mask_input_as_output_without_sam=True,
-        directly_add_no_mem_embed=True,
-        use_high_res_features_in_sam=True,
-        multimask_output_in_sam=True,
-        iou_prediction_use_sigmoid=True,
-        use_obj_ptrs_in_encoder=True,
-        add_tpos_enc_to_obj_ptrs=True,
-        only_obj_ptrs_in_the_past_for_eval=True,
-        pred_obj_scores=True,
-        pred_obj_scores_mlp=True,
-        fixed_no_obj_ptr=True,
-        multimask_output_for_tracking=True,
-        use_multimask_token_for_obj_ptr=True,
-        multimask_min_pt_num=0,
-        multimask_max_pt_num=1,
-        use_mlp_for_obj_ptr_proj=True,
-        compile_image_encoder=False,
-        sam_mask_decoder_extra_args=dict(
-            dynamic_multimask_via_stability=True,
-            dynamic_multimask_stability_delta=0.05,
-            dynamic_multimask_stability_thresh=0.98,
-        ),
-    )
-
-    if checkpoint is not None:
-        checkpoint = attempt_download_asset(checkpoint)
-        with open(checkpoint, "rb") as f:
-            state_dict = torch.load(f)["model"]
-        sam2.load_state_dict(state_dict)
-    sam2.eval()
-    return sam2
-
-
-sam_model_map = {
-    "sam2_t.pt": build_sam2_t,
-    "sam2_s.pt": build_sam2_s,
-    "sam2_b.pt": build_sam2_b,
-    "sam2_l.pt": build_sam2_l,
-}
-
-
-def build_sam2(ckpt="sam_b.pt"):
-    """Constructs a Segment Anything Model (SAM2) based on the specified checkpoint, with various size options."""
-    model_builder = None
-    ckpt = str(ckpt)  # to allow Path ckpt types
-    for k in sam_model_map.keys():
-        if ckpt.endswith(k):
-            model_builder = sam_model_map.get(k)
-
-    if not model_builder:
-        raise FileNotFoundError(f"{ckpt} is not a supported SAM model. Available models are: \n {sam_model_map.keys()}")
-
-    return model_builder(ckpt)

ultralytics/models/sam2/model.py

@@ -1,97 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
-"""
-SAM2 model interface.
-
-This module provides an interface to the Segment Anything Model (SAM2) from Ultralytics, designed for real-time image
-segmentation tasks. The SAM2 model allows for promptable segmentation with unparalleled versatility in image analysis,
-and has been trained on the SA-1B dataset. It features zero-shot performance capabilities, enabling it to adapt to new
-image distributions and tasks without prior knowledge.
-
-Key Features:
-    - Promptable segmentation
-    - Real-time performance
-    - Zero-shot transfer capabilities
-    - Trained on SA-1B dataset
-"""
-
-from ultralytics.models.sam import SAM
-
-from .build import build_sam2
-from .predict import SAM2Predictor
-
-
-class SAM2(SAM):
-    """
-    SAM2 class for real-time image segmentation using the Segment Anything Model (SAM2).
-
-    This class extends the SAM base class, providing an interface to the SAM2 model for promptable segmentation
-    tasks. It supports loading pre-trained weights and offers zero-shot performance capabilities.
-
-    Attributes:
-        model (torch.nn.Module): The loaded SAM2 model.
-        task_map (Dict[str, Type[SAM2Predictor]]): Mapping of 'segment' task to SAM2Predictor.
-
-    Methods:
-        __init__: Initializes the SAM2 model with pre-trained weights.
-        _load: Loads specified weights into the SAM2 model.
-
-    Examples:
-        >>> sam2 = SAM2("sam2_b.pt")
-        >>> sam2._load('path/to/sam2_weights.pt')
-        >>> task_map = sam2.task_map
-        >>> print(task_map)
-        {'segment': SAM2Predictor}
-
-    Notes:
-        - Supports .pt and .pth file extensions for model weights.
-        - Offers zero-shot transfer capabilities for new image distributions and tasks.
-    """
-
-    def __init__(self, model="sam2_b.pt") -> None:
-        """
-        Initializes the SAM2 model with a pre-trained model file.
-
-        Args:
-            model (str): Path to the pre-trained SAM2 model file. File should have a .pt or .pth extension.
-
-        Raises:
-            NotImplementedError: If the model file extension is not .pt or .pth.
-
-        Examples:
-            >>> sam2 = SAM2("sam2_b.pt")
-        """
-        super().__init__(model=model)
-
-    def _load(self, weights: str, task=None):
-        """
-        Loads the specified weights into the SAM2 model.
-
-        This method is responsible for loading pre-trained weights into the SAM2 model. It supports loading
-        weights from files with .pt or .pth extensions.
-
-        Args:
-            weights (str): Path to the weights file. Should be a file with .pt or .pth extension.
-            task (str | None): Task name. If provided, it may be used to configure model-specific settings.
-
-        Examples:
-            >>> sam2_model = SAM2()
-            >>> sam2_model._load('path/to/sam2_weights.pt')
-        """
-        self.model = build_sam2(weights)
-
-    @property
-    def task_map(self):
-        """
-        Provides a mapping from the 'segment' task to its corresponding 'Predictor'.
-
-        Returns:
-            (Dict[str, Type[SAM2Predictor]]): A dictionary mapping the 'segment' task to its corresponding
-                SAM2Predictor class.
-
-        Examples:
-            >>> sam2 = SAM2()
-            >>> task_map = sam2.task_map
-            >>> print(task_map)
-            {'segment': SAM2Predictor}
-        """
-        return {"segment": {"predictor": SAM2Predictor}}

ultralytics/models/sam2/modules/__init__.py

@@ -1 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license