ultralytics 8.3.10__py3-none-any.whl → 8.3.12__py3-none-any.whl

{ultralytics-8.3.10.dist-info → ultralytics-8.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.3.10
+Version: 8.3.12
 Summary: Ultralytics YOLO for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -60,10 +60,6 @@ Requires-Dist: mkdocs-jupyter; extra == "dev"
 Requires-Dist: mkdocs-redirects; extra == "dev"
 Requires-Dist: mkdocs-ultralytics-plugin>=0.1.8; extra == "dev"
 Requires-Dist: mkdocs-macros-plugin>=1.0.5; extra == "dev"
-Provides-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"
@@ -85,6 +81,9 @@ Provides-Extra: logging
 Requires-Dist: comet; extra == "logging"
 Requires-Dist: tensorboard>=2.13.0; extra == "logging"
 Requires-Dist: dvclive>=2.12.0; extra == "logging"
+Provides-Extra: solutions
+Requires-Dist: shapely>=2.0.0; extra == "solutions"
+Requires-Dist: streamlit; extra == "solutions"
 
 <div align="center">
   <p>

{ultralytics-8.3.10.dist-info → ultralytics-8.3.12.dist-info}/RECORD

@@ -1,16 +1,16 @@
 tests/__init__.py,sha256=iVH5nXrACTDv0_ZIVRPi-9f6oYBl6g-tCkeR2Hb8MFM,666
 tests/conftest.py,sha256=9PFAiwAy6eeORGspr5dOKxVuFDVKqYg8Nn_RxSJ27UI,2919
-tests/test_cli.py,sha256=E4lMt49TGo12Lb5CgQfpk1bwyFUZuFxF0V9j_ykV7xM,4821
-tests/test_cuda.py,sha256=KoRtRLUB7KOb9IXYX4mCi295Uh_cZEEFhCyvCDGRK9s,5381
+tests/test_cli.py,sha256=G7OJ1ErQYsGy2Dx1zP-0p7EZR4aPoAdtLGiY4Hm7jQM,5006
+tests/test_cuda.py,sha256=cxohrqAjjPCgTaChYhQ4CObQzz7ZyhVGPTxkcxHSf0g,5940
 tests/test_engine.py,sha256=dcEcJsMQh61rDSNv7l4TIAgybLpzjVwerv9JZC_KCM8,4934
 tests/test_exports.py,sha256=fpTKEVBUGLF3WiZPNKRs-IEcIY4cfxgvgKjUNfodjww,8042
 tests/test_integrations.py,sha256=f5-QCUk1SU_-qn4mBCZwS3GN3tXEBIIXo4z2EhExbHw,6126
 tests/test_python.py,sha256=I1RRdCwLdrc3jX06huVxct8HX8ccQOmQgVpuEflRl0U,23560
 tests/test_solutions.py,sha256=dpxWGKO-aJ3Yff4KR7BQGajX9VyFdGTWEtcbmFC3WwE,3005
-ultralytics/__init__.py,sha256=r6LeW7qfMLanc7g2MYd3t48Oqt6LLDRB_PJFMVyyK3E,753
+ultralytics/__init__.py,sha256=7skXKgnCJr4mr4nr_Cg_IOl3PvRxQ_x4K7AHcxnor2Y,681
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
-ultralytics/cfg/__init__.py,sha256=N-XONBXwmD3vzoE4icBXznkV8LOLmf6ak6mRdGPucvw,33146
+ultralytics/cfg/__init__.py,sha256=CUg1z4zY3KyR-V4_bghMY8s1xuu-M50gm-v_vpHdXEM,31753
 ultralytics/cfg/default.yaml,sha256=ul49zgSzTegMmc8CFeu9tXkWNvQhETdZMa9EgDNSnY4,8319
 ultralytics/cfg/datasets/Argoverse.yaml,sha256=FyeuJT5CHq_9d4hlfAf0kpZlnbUMO0S--UJ1yIqcdKk,3134
 ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=QVfp_Qp-4rukuicaB4qx86NxSHM8Mrzym8l_fIDo8gw,1195
@@ -85,11 +85,11 @@ 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/solutions/default.yaml,sha256=CByxINYMyoGzGKdurDk2GhYc8XOa8Z6H7CZx7uZSPpc,1532
+ultralytics/cfg/solutions/default.yaml,sha256=zZ_ksoZ-BcAIL5jjw0jzHgraoe7363oxXqfSsg9yopk,1673
 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=PniOxH2MScWKp539vuufk69uG1JsltDB5OMCUhxn2QY,2489
+ultralytics/data/annotator.py,sha256=oy87bzQN6ZRYeucoLk8e-jDEo6YJ91FE_zMFtLEVC1I,2489
 ultralytics/data/augment.py,sha256=YCLrwx1mRGeidggo_7GeINay8KdxACqREHJofZeaTHA,120430
 ultralytics/data/base.py,sha256=ZCIhAyFfxXVp5fVnYD8mwbksNALJTayBKIR5FKGV7ZM,15168
 ultralytics/data/build.py,sha256=AfMmz0sHIYmwry_90tEJFRk_kz0S3SolScVXqYHiT08,7261
@@ -98,18 +98,13 @@ ultralytics/data/dataset.py,sha256=D556AW0ZEsW3V8c5zJiHM_prc_YfZqymIkDKPw3k9Io,2
 ultralytics/data/loaders.py,sha256=Fr70Q9p9t7buLW_8R2_lI_nyCMG033gWSxvwy1M-a-U,28449
 ultralytics/data/split_dota.py,sha256=yOtypHoY5HvIVBKZgFXdfj2tuCLLEBnMwNfAeG94Eik,10680
 ultralytics/data/utils.py,sha256=u6OZ7InLpI1em5aEPz13ZzS9BcO37dcY9_s2btXGZYQ,31076
-ultralytics/data/explorer/__init__.py,sha256=-Y3m1ZedepOQUv_KW82zaGxvU_PSHcuwUTFqG9BhAr4,113
-ultralytics/data/explorer/explorer.py,sha256=JWmLHHhp68h2q3vx4poBou5RYoAX3R89yihR50YLDb0,18881
-ultralytics/data/explorer/utils.py,sha256=EvvukQiQUTBrsZznmMnyEX2EqTuwZo_Geyc8yfi8NIA,7085
-ultralytics/data/explorer/gui/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
-ultralytics/data/explorer/gui/dash.py,sha256=6XOZy9NrkPEXREJPbi0EBkGgu78TAdHpdhSB2HuBOAo,10222
 ultralytics/engine/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
-ultralytics/engine/exporter.py,sha256=DeHW_T_Zd3A21BLQYV1-FnS5EcmepMOy9nrussYNieU,57505
-ultralytics/engine/model.py,sha256=Vtkza7cQrxvowb0PqGFhp7eC3cXRIKj6OUaR5d9w1-U,51464
+ultralytics/engine/exporter.py,sha256=AWhGvR0c-8NiId6pt5hNpwnOPZ03DIIdBTlh1JkqBHs,57669
+ultralytics/engine/model.py,sha256=pvL1uf-wwdWL8Iph7VEAYn1-z7wEHzVug21V_0_gO6M,51456
 ultralytics/engine/predictor.py,sha256=keTelEeo23Dcbs-XvmRWAPIs4pbCNDtsMBz88WM1eK8,17534
-ultralytics/engine/results.py,sha256=8RJlN8J-_9w-mrDZm9wC-DZJTPBS7v1c_r_R173QyRM,75043
+ultralytics/engine/results.py,sha256=BxanBI8PhBCfs-9cSy-GS6naScuiD3hdvUAJWPW2mS0,75043
 ultralytics/engine/trainer.py,sha256=6dGOEZvMo3o97SLpKlcR5XmhWhUHh05uLYpj3jNn0jU,36981
-ultralytics/engine/tuner.py,sha256=gPqDTHH7vRB2O3YyH26m1BjVKbXxuA2XAlPRzTKFZsc,11838
+ultralytics/engine/tuner.py,sha256=WBj8iw1K1TK0hvanlA-wkwmfqh1SI8jEe2dGwUINeTg,11838
 ultralytics/engine/validator.py,sha256=2C_qXI36Z9rLOpmS0YR8Qe3ka4p23YiH2w5ai7-XBwE,14811
 ultralytics/hub/__init__.py,sha256=3SKvZ5aRina3h94xMPQIB3D4maF62qFcyIqPPHRHNAc,5644
 ultralytics/hub/auth.py,sha256=kDLakGa2NbzvMAeXc2UdzZ65r0AH-XeM_JfsDY97WGk,5545
@@ -135,7 +130,7 @@ ultralytics/models/sam/__init__.py,sha256=o4_D6y8YJlOXIK7Lwo9RHnIJJ9xoFNi4zK99QS
 ultralytics/models/sam/amg.py,sha256=GrmO_8YfIDt_QkPEMF_WFjPZkhwhf7iwx7ig8JgOUnE,8709
 ultralytics/models/sam/build.py,sha256=np9vP7AETCZA2Wdds-uj2eQKVnpHQaVpRrE2-U2uMTI,12153
 ultralytics/models/sam/model.py,sha256=2KFUp8SHiqOgwUjkdqdau0oduJwKQxm4N9GHWjdhUFo,7382
-ultralytics/models/sam/predict.py,sha256=_spP0uYNFzUnybwBvzZhF3iEMwvAi6bxryRdUwxwweM,38608
+ultralytics/models/sam/predict.py,sha256=tvr-CHlk5yyKlt_UhozEruXkD1pheNZA2dC3PCT6nMY,39001
 ultralytics/models/sam/modules/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
 ultralytics/models/sam/modules/blocks.py,sha256=Q-KwhFbdyZhl1tjG_kP2LcQkZbzoNt618i-NRrKNx2Y,45919
 ultralytics/models/sam/modules/decoders.py,sha256=mODsqnTN_CjE3H0Sh9cd8PfTnHANPjGB1bjqHxfezSg,25830
@@ -149,15 +144,15 @@ ultralytics/models/utils/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8
 ultralytics/models/utils/loss.py,sha256=Ozi0Up7cmPAehXgqGuVSAtUS4XQxaX76KqE8Q0VHk7E,15840
 ultralytics/models/utils/ops.py,sha256=aPAPwWMLJLWq-I04wS_YrqJ_Vy_xBXtqQu6Aox15YDA,13221
 ultralytics/models/yolo/__init__.py,sha256=e1cZr9pbSbf3Ya2OvkTjGRwD_E2YZpe610xskBM8gEk,247
-ultralytics/models/yolo/model.py,sha256=CMh3_qYCm2mdFHVyZJDMu8eFCTMD0z1ZPmM8GmwTU7E,4233
+ultralytics/models/yolo/model.py,sha256=E4TuJZZux0L_SG7sC0SDgxrmeBvuZRpxprPrCC26lvs,4233
 ultralytics/models/yolo/classify/__init__.py,sha256=t-4pUHmgI2gjhc-l3bqNEcEtKD1dO40nD4Vc6Y2xD6o,355
 ultralytics/models/yolo/classify/predict.py,sha256=0CEJ4B4fXbOMUnJy79gRvG-qdszOzTSLOb1xxkgsKek,2444
 ultralytics/models/yolo/classify/train.py,sha256=THXSkQVQVBuw1QxcEVA8MtLHYYdaAEqepObJCXoLcZ8,6358
 ultralytics/models/yolo/classify/val.py,sha256=Tzizhp3ebzPvwJejrE8tb-TuXw4MdkEI9mOANV74eXQ,4909
 ultralytics/models/yolo/detect/__init__.py,sha256=JR8gZJWn7wMBbh-0j_073nxJVZTMFZVWTOG5Wnvk6w0,229
-ultralytics/models/yolo/detect/predict.py,sha256=_kY6-_wsPCt9ZOf-iwusceikAM5TV_KnjYdv2koE45A,1471
-ultralytics/models/yolo/detect/train.py,sha256=cHlCTj39Tr6butQOQRreCFnVODK2IqaogBpTAQuGAPE,6363
-ultralytics/models/yolo/detect/val.py,sha256=qZDP1ETpZE1e7avZyOuBe3U8zuLpsBdKHUuN3YL-tQI,15152
+ultralytics/models/yolo/detect/predict.py,sha256=-uZFLutxGYZX47RANcaxC-LFStRbv0nBv_8-ypadQoI,1471
+ultralytics/models/yolo/detect/train.py,sha256=jayPByykRhJvUeDCN-oZ4M6Yx0rpjAWq9epjJgnHeYA,6363
+ultralytics/models/yolo/detect/val.py,sha256=ulKYfViVU2WErDtFOU8FTGBlROR4p-jFK4Nf9O4LMV4,15152
 ultralytics/models/yolo/obb/__init__.py,sha256=txWbPGLY1_M7ZwlLQjrwGjTBOlsv9P3yk5ZEgysTinU,193
 ultralytics/models/yolo/obb/predict.py,sha256=VxpKCKV5dWnOr0GyV1rJGH5SzzRouCYW_8T26xJ8MU8,2037
 ultralytics/models/yolo/obb/train.py,sha256=_FVYCvHJ5ECi2aN8k7AmVLxRUuun7acSqwWtCBRuL6Q,1473
@@ -174,7 +169,7 @@ ultralytics/models/yolo/world/__init__.py,sha256=3VTH0q4NOt2EWRom15yCymvmvm0Etp2
 ultralytics/models/yolo/world/train.py,sha256=gaDrAmLJpg9qDtmL5evA5HsV2yb4RTRSfk2EDYrHdRg,3686
 ultralytics/models/yolo/world/train_world.py,sha256=IsnCEVt6DcM9lUskCKmIN-M8MM79xLpwTRqRoAHUnZ4,4857
 ultralytics/nn/__init__.py,sha256=4BPLHY89xEM_al5uK0aOmFgiML6CMGEZbezxOvTjOEs,587
-ultralytics/nn/autobackend.py,sha256=aBW_Z8XnSsD-vW7Ek873dyKX9h55XHIYwTG22M3eOIk,31599
+ultralytics/nn/autobackend.py,sha256=xcbndT-esNu3Icx3SWRJXZ7JqWjj8H5hteUetYBliDo,31599
 ultralytics/nn/tasks.py,sha256=ssBZR4LY4rvaxYawXq5-yWSBAZ9oCz6BgxWYXB2YD68,48399
 ultralytics/nn/modules/__init__.py,sha256=xhW2BennT9U_VaMXVpRu-bdLgp1BXt9L8mkIUBE3idU,2625
 ultralytics/nn/modules/activation.py,sha256=chhn469wnRHEs5BMGNBYXwPYZc_7-urspTT8fnBd-xA,895
@@ -185,7 +180,7 @@ ultralytics/nn/modules/transformer.py,sha256=tGiK8NmPfswwW1rbF21r5ILUkkZQ6Nk4s8j
 ultralytics/nn/modules/utils.py,sha256=a88cKl2wz1nMVSEBiajtvaCbDBQIkESWOKTZ_WAJy90,3195
 ultralytics/solutions/__init__.py,sha256=6RDeXWO1QSaMgCq8YrWXaj2xvPw2sJwJL_a0dgjCvz0,648
 ultralytics/solutions/ai_gym.py,sha256=lBAkWV8vrEdKAXcBFVbugPeZZ08MOjGYTdnFlG22vKM,3772
-ultralytics/solutions/analytics.py,sha256=bGuZes11D7DNiTsHdwu6PJ0QA0vCiqMMAtZ7NyEkshY,11568
+ultralytics/solutions/analytics.py,sha256=w5hnnBNSTQ35tJp6DDeWYw2ASjylp3ZmzrTXcdWwDw8,9319
 ultralytics/solutions/distance_calculation.py,sha256=o_DAHk4JX8n2Vt7E68MX67mREOBZuy5skbXtVZ6iu_4,5228
 ultralytics/solutions/heatmap.py,sha256=2C4s_rVFcOc5oSWxb0pNxNoCawe4lxajpTDNFd4tVL8,3850
 ultralytics/solutions/object_counter.py,sha256=1Nsivk-cyGBM1G6eWe11_vdDWTdbJwaUFMJ1A7OK-Qg,5495
@@ -205,12 +200,12 @@ ultralytics/trackers/utils/kalman_filter.py,sha256=cH9zD3fwkuezP97H9mw8cSBN7a8hH
 ultralytics/trackers/utils/matching.py,sha256=3Ie1WNNRZ4_q3365F03XD7Nr9juZB_08mw4yUKC3w74,7162
 ultralytics/utils/__init__.py,sha256=du1Y1LMU0jQn_zWWnAIx9U8wn6Vh7ce-k7qMwi6y0po,48698
 ultralytics/utils/autobatch.py,sha256=BO9MCRtrLDtrDQaxqV0BdjaYsgXf-q07Y3_VdGp4URY,4330
-ultralytics/utils/benchmarks.py,sha256=8FYp5WPzcxcDaeg8ol2sgzRBHVGYatEO7f3MrmPF6nI,25097
-ultralytics/utils/checks.py,sha256=SsB3s1z9TtMjGelDkGZIi6B40VXmCtGw2hcOCyPikx4,29765
+ultralytics/utils/benchmarks.py,sha256=R3_jtwLd48azPSXmtIhqlzduUvflk0FOY8GJOJ6mB6E,25097
+ultralytics/utils/checks.py,sha256=iH5R-DQKhP7qnW8pQm-rlYPsMJ5KWvcn9imo-hamIqE,29765
 ultralytics/utils/dist.py,sha256=NDFga-uKxkBX2zLxFHSene_cCiGQJoyOeCXcN9JIOIk,2358
-ultralytics/utils/downloads.py,sha256=o8RY9f0KrzWfueLs8DuJ5w8OWQ-ll4ZS9lX6MEFDi70,21977
+ultralytics/utils/downloads.py,sha256=xxM9zTNCqo3PRFOxWCH1pJoKuFJKZFsb3sab6Hvabug,21977
 ultralytics/utils/errors.py,sha256=GqP_Jgj_n0paxn8OMhn3DTCgoNkB2WjUcUaqs-M6SQk,816
-ultralytics/utils/files.py,sha256=YjfzbBDAq-nD3LKjtuMVwggnnv1dROMuVoo3Edm_tjU,8224
+ultralytics/utils/files.py,sha256=uiXQSVABJRoI5ImnM6ndEBIFbECfksmWNEldBg8GnSo,8224
 ultralytics/utils/instance.py,sha256=QSms7mPHZ5e8JGuJYLohLWltzI0aBE8dob2rOUK4RtM,16249
 ultralytics/utils/loss.py,sha256=SW3FVFFp8Ki_LCT8wIdFbm6KmyPcQn3RmKNcvVAhMQI,34174
 ultralytics/utils/metrics.py,sha256=UgLGudWp57uXDMlMUJy4gsz6cfVjcq7tYmHeto3TqvM,53927
@@ -220,7 +215,7 @@ ultralytics/utils/plotting.py,sha256=aozAEwcbc447ume9bQrEBTU04AzyiZZrnzcTzA2S6j0
 ultralytics/utils/tal.py,sha256=ECsu95xEqOItmxMDN4YTD3FsUiIsQNWy0pZC3TfvFfk,16877
 ultralytics/utils/torch_utils.py,sha256=gVN-KSrAzJC1rW3woQd4FsTT693GD8rXiccToL2m4kM,30059
 ultralytics/utils/triton.py,sha256=gg1finxno_tY2Ge9PMhmu7PI9wvoFZoiicdT4Bhqv3w,3936
-ultralytics/utils/tuner.py,sha256=AtEtK6pOt9xVTyx864OpNRVxNdAxz5aKHzveiXwkD1A,6250
+ultralytics/utils/tuner.py,sha256=mJdgvuE2StoFS13mEdsTbsxQgSZA4fSdSCgoyh8PvNw,6250
 ultralytics/utils/callbacks/__init__.py,sha256=YrWqC3BVVaTLob4iCPR6I36mUxIUOpPJW7B_LjT78Qw,214
 ultralytics/utils/callbacks/base.py,sha256=PHjQ6RITwC2dylCQTB0bdPgAsHjxVeuDb5N1NPTbHGc,5775
 ultralytics/utils/callbacks/clearml.py,sha256=qbLbqzMVWAnjqg5YUM-Ue6CmGueFCvqKpHFKlw-MyVc,5933
@@ -232,9 +227,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=IbGQfEltamUKXJt93uSLQFn8c2rYh3DMTg
 ultralytics/utils/callbacks/raytune.py,sha256=ODVYzy-CoM4Uge0zjkh3Hnh9nF2M0vhDrSenXnvcizw,705
 ultralytics/utils/callbacks/tensorboard.py,sha256=bv4fkkesdgmZv_E2MU6wuaMBwEV5iI2G53RHPyD9quw,4170
 ultralytics/utils/callbacks/wb.py,sha256=upfbF8-LLXueUvulLaMDmKDhKCl_PWbNa_87PQ0L0Rc,6752
-ultralytics-8.3.10.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
-ultralytics-8.3.10.dist-info/METADATA,sha256=erZGLlFck6gorIKxGLLR-ymgpHCb5WiGGa89PyM_sQs,34700
-ultralytics-8.3.10.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-ultralytics-8.3.10.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
-ultralytics-8.3.10.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-ultralytics-8.3.10.dist-info/RECORD,,
+ultralytics-8.3.12.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
+ultralytics-8.3.12.dist-info/METADATA,sha256=nuBjn3vHHBwnaqu1b8fskjmJDCrSUQCTsPwJOceuytg,34660
+ultralytics-8.3.12.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ultralytics-8.3.12.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
+ultralytics-8.3.12.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+ultralytics-8.3.12.dist-info/RECORD,,

ultralytics/data/explorer/__init__.py

@@ -1,5 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
-
-from .utils import plot_query_result
-
-__all__ = ["plot_query_result"]

ultralytics/data/explorer/explorer.py

@@ -1,460 +0,0 @@
-# Ultralytics YOLO 🚀, AGPL-3.0 license
-
-from io import BytesIO
-from pathlib import Path
-from typing import Any, List, Tuple, Union
-
-import cv2
-import numpy as np
-import torch
-from matplotlib import pyplot as plt
-from PIL import Image
-from tqdm import tqdm
-
-from ultralytics.data.augment import Format
-from ultralytics.data.dataset import YOLODataset
-from ultralytics.data.utils import check_det_dataset
-from ultralytics.models.yolo.model import YOLO
-from ultralytics.utils import LOGGER, USER_CONFIG_DIR, IterableSimpleNamespace, checks
-
-from .utils import get_sim_index_schema, get_table_schema, plot_query_result, prompt_sql_query, sanitize_batch
-
-
-class ExplorerDataset(YOLODataset):
-    """Extends YOLODataset for advanced data exploration and manipulation in model training workflows."""
-
-    def __init__(self, *args, data: dict = None, **kwargs) -> None:
-        """Initializes the ExplorerDataset with the provided data arguments, extending the YOLODataset class."""
-        super().__init__(*args, data=data, **kwargs)
-
-    def load_image(self, i: int) -> Union[Tuple[np.ndarray, Tuple[int, int], Tuple[int, int]], Tuple[None, None, None]]:
-        """Loads 1 image from dataset index 'i' without any resize ops."""
-        im, f, fn = self.ims[i], self.im_files[i], self.npy_files[i]
-        if im is None:  # not cached in RAM
-            if fn.exists():  # load npy
-                im = np.load(fn)
-            else:  # read image
-                im = cv2.imread(f)  # BGR
-                if im is None:
-                    raise FileNotFoundError(f"Image Not Found {f}")
-            h0, w0 = im.shape[:2]  # orig hw
-            return im, (h0, w0), im.shape[:2]
-
-        return self.ims[i], self.im_hw0[i], self.im_hw[i]
-
-    def build_transforms(self, hyp: IterableSimpleNamespace = None):
-        """Creates transforms for dataset images without resizing."""
-        return Format(
-            bbox_format="xyxy",
-            normalize=False,
-            return_mask=self.use_segments,
-            return_keypoint=self.use_keypoints,
-            batch_idx=True,
-            mask_ratio=hyp.mask_ratio,
-            mask_overlap=hyp.overlap_mask,
-        )
-
-
-class Explorer:
-    """Utility class for image embedding, table creation, and similarity querying using LanceDB and YOLO models."""
-
-    def __init__(
-        self,
-        data: Union[str, Path] = "coco128.yaml",
-        model: str = "yolov8n.pt",
-        uri: str = USER_CONFIG_DIR / "explorer",
-    ) -> None:
-        """Initializes the Explorer class with dataset path, model, and URI for database connection."""
-        # Note duckdb==0.10.0 bug https://github.com/ultralytics/ultralytics/pull/8181
-        checks.check_requirements(["lancedb>=0.4.3", "duckdb<=0.9.2"])
-        import lancedb
-
-        self.connection = lancedb.connect(uri)
-        self.table_name = f"{Path(data).name.lower()}_{model.lower()}"
-        self.sim_idx_base_name = (
-            f"{self.table_name}_sim_idx".lower()
-        )  # Use this name and append thres and top_k to reuse the table
-        self.model = YOLO(model)
-        self.data = data  # None
-        self.choice_set = None
-
-        self.table = None
-        self.progress = 0
-
-    def create_embeddings_table(self, force: bool = False, split: str = "train") -> None:
-        """
-        Create LanceDB table containing the embeddings of the images in the dataset. The table will be reused if it
-        already exists. Pass force=True to overwrite the existing table.
-
-        Args:
-            force (bool): Whether to overwrite the existing table or not. Defaults to False.
-            split (str): Split of the dataset to use. Defaults to 'train'.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            ```
-        """
-        if self.table is not None and not force:
-            LOGGER.info("Table already exists. Reusing it. Pass force=True to overwrite it.")
-            return
-        if self.table_name in self.connection.table_names() and not force:
-            LOGGER.info(f"Table {self.table_name} already exists. Reusing it. Pass force=True to overwrite it.")
-            self.table = self.connection.open_table(self.table_name)
-            self.progress = 1
-            return
-        if self.data is None:
-            raise ValueError("Data must be provided to create embeddings table")
-
-        data_info = check_det_dataset(self.data)
-        if split not in data_info:
-            raise ValueError(
-                f"Split {split} is not found in the dataset. Available keys in the dataset are {list(data_info.keys())}"
-            )
-
-        choice_set = data_info[split]
-        choice_set = choice_set if isinstance(choice_set, list) else [choice_set]
-        self.choice_set = choice_set
-        dataset = ExplorerDataset(img_path=choice_set, data=data_info, augment=False, cache=False, task=self.model.task)
-
-        # Create the table schema
-        batch = dataset[0]
-        vector_size = self.model.embed(batch["im_file"], verbose=False)[0].shape[0]
-        table = self.connection.create_table(self.table_name, schema=get_table_schema(vector_size), mode="overwrite")
-        table.add(
-            self._yield_batches(
-                dataset,
-                data_info,
-                self.model,
-                exclude_keys=["img", "ratio_pad", "resized_shape", "ori_shape", "batch_idx"],
-            )
-        )
-
-        self.table = table
-
-    def _yield_batches(self, dataset: ExplorerDataset, data_info: dict, model: YOLO, exclude_keys: List[str]):
-        """Generates batches of data for embedding, excluding specified keys."""
-        for i in tqdm(range(len(dataset))):
-            self.progress = float(i + 1) / len(dataset)
-            batch = dataset[i]
-            for k in exclude_keys:
-                batch.pop(k, None)
-            batch = sanitize_batch(batch, data_info)
-            batch["vector"] = model.embed(batch["im_file"], verbose=False)[0].detach().tolist()
-            yield [batch]
-
-    def query(
-        self, imgs: Union[str, np.ndarray, List[str], List[np.ndarray]] = None, limit: int = 25
-    ) -> Any:  # pyarrow.Table
-        """
-        Query the table for similar images. Accepts a single image or a list of images.
-
-        Args:
-            imgs (str or list): Path to the image or a list of paths to the images.
-            limit (int): Number of results to return.
-
-        Returns:
-            (pyarrow.Table): An arrow table containing the results. Supports converting to:
-                - pandas dataframe: `result.to_pandas()`
-                - dict of lists: `result.to_pydict()`
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            similar = exp.query(img="https://ultralytics.com/images/zidane.jpg")
-            ```
-        """
-        if self.table is None:
-            raise ValueError("Table is not created. Please create the table first.")
-        if isinstance(imgs, str):
-            imgs = [imgs]
-        assert isinstance(imgs, list), f"img must be a string or a list of strings. Got {type(imgs)}"
-        embeds = self.model.embed(imgs)
-        # Get avg if multiple images are passed (len > 1)
-        embeds = torch.mean(torch.stack(embeds), 0).cpu().numpy() if len(embeds) > 1 else embeds[0].cpu().numpy()
-        return self.table.search(embeds).limit(limit).to_arrow()
-
-    def sql_query(
-        self, query: str, return_type: str = "pandas"
-    ) -> Union[Any, None]:  # pandas.DataFrame or pyarrow.Table
-        """
-        Run a SQL-Like query on the table. Utilizes LanceDB predicate pushdown.
-
-        Args:
-            query (str): SQL query to run.
-            return_type (str): Type of the result to return. Can be either 'pandas' or 'arrow'. Defaults to 'pandas'.
-
-        Returns:
-            (pyarrow.Table): An arrow table containing the results.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            query = "SELECT * FROM 'table' WHERE labels LIKE '%person%'"
-            result = exp.sql_query(query)
-            ```
-        """
-        assert return_type in {
-            "pandas",
-            "arrow",
-        }, f"Return type should be either `pandas` or `arrow`, but got {return_type}"
-        import duckdb
-
-        if self.table is None:
-            raise ValueError("Table is not created. Please create the table first.")
-
-        # Note: using filter pushdown would be a better long term solution. Temporarily using duckdb for this.
-        table = self.table.to_arrow()  # noqa NOTE: Don't comment this. This line is used by DuckDB
-        if not query.startswith("SELECT") and not query.startswith("WHERE"):
-            raise ValueError(
-                f"Query must start with SELECT or WHERE. You can either pass the entire query or just the WHERE "
-                f"clause. found {query}"
-            )
-        if query.startswith("WHERE"):
-            query = f"SELECT * FROM 'table' {query}"
-        LOGGER.info(f"Running query: {query}")
-
-        rs = duckdb.sql(query)
-        if return_type == "arrow":
-            return rs.arrow()
-        elif return_type == "pandas":
-            return rs.df()
-
-    def plot_sql_query(self, query: str, labels: bool = True) -> Image.Image:
-        """
-        Plot the results of a SQL-Like query on the table.
-
-        Args:
-            query (str): SQL query to run.
-            labels (bool): Whether to plot the labels or not.
-
-        Returns:
-            (PIL.Image): Image containing the plot.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            query = "SELECT * FROM 'table' WHERE labels LIKE '%person%'"
-            result = exp.plot_sql_query(query)
-            ```
-        """
-        result = self.sql_query(query, return_type="arrow")
-        if len(result) == 0:
-            LOGGER.info("No results found.")
-            return None
-        img = plot_query_result(result, plot_labels=labels)
-        return Image.fromarray(img)
-
-    def get_similar(
-        self,
-        img: Union[str, np.ndarray, List[str], List[np.ndarray]] = None,
-        idx: Union[int, List[int]] = None,
-        limit: int = 25,
-        return_type: str = "pandas",
-    ) -> Any:  # pandas.DataFrame or pyarrow.Table
-        """
-        Query the table for similar images. Accepts a single image or a list of images.
-
-        Args:
-            img (str or list): Path to the image or a list of paths to the images.
-            idx (int or list): Index of the image in the table or a list of indexes.
-            limit (int): Number of results to return. Defaults to 25.
-            return_type (str): Type of the result to return. Can be either 'pandas' or 'arrow'. Defaults to 'pandas'.
-
-        Returns:
-            (pandas.DataFrame): A dataframe containing the results.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            similar = exp.get_similar(img="https://ultralytics.com/images/zidane.jpg")
-            ```
-        """
-        assert return_type in {"pandas", "arrow"}, f"Return type should be `pandas` or `arrow`, but got {return_type}"
-        img = self._check_imgs_or_idxs(img, idx)
-        similar = self.query(img, limit=limit)
-
-        if return_type == "arrow":
-            return similar
-        elif return_type == "pandas":
-            return similar.to_pandas()
-
-    def plot_similar(
-        self,
-        img: Union[str, np.ndarray, List[str], List[np.ndarray]] = None,
-        idx: Union[int, List[int]] = None,
-        limit: int = 25,
-        labels: bool = True,
-    ) -> Image.Image:
-        """
-        Plot the similar images. Accepts images or indexes.
-
-        Args:
-            img (str or list): Path to the image or a list of paths to the images.
-            idx (int or list): Index of the image in the table or a list of indexes.
-            labels (bool): Whether to plot the labels or not.
-            limit (int): Number of results to return. Defaults to 25.
-
-        Returns:
-            (PIL.Image): Image containing the plot.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            similar = exp.plot_similar(img="https://ultralytics.com/images/zidane.jpg")
-            ```
-        """
-        similar = self.get_similar(img, idx, limit, return_type="arrow")
-        if len(similar) == 0:
-            LOGGER.info("No results found.")
-            return None
-        img = plot_query_result(similar, plot_labels=labels)
-        return Image.fromarray(img)
-
-    def similarity_index(self, max_dist: float = 0.2, top_k: float = None, force: bool = False) -> Any:  # pd.DataFrame
-        """
-        Calculate the similarity index of all the images in the table. Here, the index will contain the data points that
-        are max_dist or closer to the image in the embedding space at a given index.
-
-        Args:
-            max_dist (float): maximum L2 distance between the embeddings to consider. Defaults to 0.2.
-            top_k (float): Percentage of the closest data points to consider when counting. Used to apply limit.
-                           vector search. Defaults: None.
-            force (bool): Whether to overwrite the existing similarity index or not. Defaults to True.
-
-        Returns:
-            (pandas.DataFrame): A dataframe containing the similarity index. Each row corresponds to an image,
-                and columns include indices of similar images and their respective distances.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            sim_idx = exp.similarity_index()
-            ```
-        """
-        if self.table is None:
-            raise ValueError("Table is not created. Please create the table first.")
-        sim_idx_table_name = f"{self.sim_idx_base_name}_thres_{max_dist}_top_{top_k}".lower()
-        if sim_idx_table_name in self.connection.table_names() and not force:
-            LOGGER.info("Similarity matrix already exists. Reusing it. Pass force=True to overwrite it.")
-            return self.connection.open_table(sim_idx_table_name).to_pandas()
-
-        if top_k and not (1.0 >= top_k >= 0.0):
-            raise ValueError(f"top_k must be between 0.0 and 1.0. Got {top_k}")
-        if max_dist < 0.0:
-            raise ValueError(f"max_dist must be greater than 0. Got {max_dist}")
-
-        top_k = int(top_k * len(self.table)) if top_k else len(self.table)
-        top_k = max(top_k, 1)
-        features = self.table.to_lance().to_table(columns=["vector", "im_file"]).to_pydict()
-        im_files = features["im_file"]
-        embeddings = features["vector"]
-
-        sim_table = self.connection.create_table(sim_idx_table_name, schema=get_sim_index_schema(), mode="overwrite")
-
-        def _yield_sim_idx():
-            """Generates a dataframe with similarity indices and distances for images."""
-            for i in tqdm(range(len(embeddings))):
-                sim_idx = self.table.search(embeddings[i]).limit(top_k).to_pandas().query(f"_distance <= {max_dist}")
-                yield [
-                    {
-                        "idx": i,
-                        "im_file": im_files[i],
-                        "count": len(sim_idx),
-                        "sim_im_files": sim_idx["im_file"].tolist(),
-                    }
-                ]
-
-        sim_table.add(_yield_sim_idx())
-        self.sim_index = sim_table
-        return sim_table.to_pandas()
-
-    def plot_similarity_index(self, max_dist: float = 0.2, top_k: float = None, force: bool = False) -> Image:
-        """
-        Plot the similarity index of all the images in the table. Here, the index will contain the data points that are
-        max_dist or closer to the image in the embedding space at a given index.
-
-        Args:
-            max_dist (float): maximum L2 distance between the embeddings to consider. Defaults to 0.2.
-            top_k (float): Percentage of closest data points to consider when counting. Used to apply limit when
-                running vector search. Defaults to 0.01.
-            force (bool): Whether to overwrite the existing similarity index or not. Defaults to True.
-
-        Returns:
-            (PIL.Image): Image containing the plot.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-
-            similarity_idx_plot = exp.plot_similarity_index()
-            similarity_idx_plot.show()  # view image preview
-            similarity_idx_plot.save("path/to/save/similarity_index_plot.png")  # save contents to file
-            ```
-        """
-        sim_idx = self.similarity_index(max_dist=max_dist, top_k=top_k, force=force)
-        sim_count = sim_idx["count"].tolist()
-        sim_count = np.array(sim_count)
-
-        indices = np.arange(len(sim_count))
-
-        # Create the bar plot
-        plt.bar(indices, sim_count)
-
-        # Customize the plot (optional)
-        plt.xlabel("data idx")
-        plt.ylabel("Count")
-        plt.title("Similarity Count")
-        buffer = BytesIO()
-        plt.savefig(buffer, format="png")
-        buffer.seek(0)
-
-        # Use Pillow to open the image from the buffer
-        return Image.fromarray(np.array(Image.open(buffer)))
-
-    def _check_imgs_or_idxs(
-        self, img: Union[str, np.ndarray, List[str], List[np.ndarray], None], idx: Union[None, int, List[int]]
-    ) -> List[np.ndarray]:
-        """Determines whether to fetch images or indexes based on provided arguments and returns image paths."""
-        if img is None and idx is None:
-            raise ValueError("Either img or idx must be provided.")
-        if img is not None and idx is not None:
-            raise ValueError("Only one of img or idx must be provided.")
-        if idx is not None:
-            idx = idx if isinstance(idx, list) else [idx]
-            img = self.table.to_lance().take(idx, columns=["im_file"]).to_pydict()["im_file"]
-
-        return img if isinstance(img, list) else [img]
-
-    def ask_ai(self, query):
-        """
-        Ask AI a question.
-
-        Args:
-            query (str): Question to ask.
-
-        Returns:
-            (pandas.DataFrame): A dataframe containing filtered results to the SQL query.
-
-        Example:
-            ```python
-            exp = Explorer()
-            exp.create_embeddings_table()
-            answer = exp.ask_ai("Show images with 1 person and 2 dogs")
-            ```
-        """
-        result = prompt_sql_query(query)
-        try:
-            return self.sql_query(result)
-        except Exception as e:
-            LOGGER.error("AI generated query is not valid. Please try again with a different prompt")
-            LOGGER.error(e)
-            return None

ultralytics/data/explorer/gui/__init__.py

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