ultralytics 8.3.148__tar.gz → 8.3.226__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (303) hide show
  1. {ultralytics-8.3.148 → ultralytics-8.3.226}/PKG-INFO +43 -36
  2. {ultralytics-8.3.148 → ultralytics-8.3.226}/README.md +26 -26
  3. {ultralytics-8.3.148 → ultralytics-8.3.226}/pyproject.toml +20 -12
  4. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/__init__.py +5 -7
  5. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/conftest.py +10 -17
  6. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_cli.py +9 -11
  7. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_cuda.py +11 -7
  8. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_engine.py +29 -2
  9. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_exports.py +82 -21
  10. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_integrations.py +11 -14
  11. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_python.py +126 -87
  12. {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_solutions.py +165 -154
  13. ultralytics-8.3.226/ultralytics/__init__.py +48 -0
  14. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/__init__.py +83 -82
  15. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/Argoverse.yaml +4 -4
  16. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/DOTAv1.5.yaml +2 -2
  17. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/DOTAv1.yaml +2 -2
  18. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/GlobalWheat2020.yaml +2 -2
  19. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/HomeObjects-3K.yaml +4 -5
  20. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/ImageNet.yaml +3 -3
  21. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/Objects365.yaml +24 -20
  22. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/SKU-110K.yaml +9 -9
  23. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/VOC.yaml +10 -13
  24. ultralytics-8.3.226/ultralytics/cfg/datasets/VisDrone.yaml +87 -0
  25. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/african-wildlife.yaml +5 -5
  26. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/brain-tumor.yaml +4 -5
  27. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/carparts-seg.yaml +5 -5
  28. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco-pose.yaml +26 -4
  29. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco.yaml +4 -4
  30. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco128-seg.yaml +2 -2
  31. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco128.yaml +2 -2
  32. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-grayscale.yaml +2 -2
  33. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-multispectral.yaml +2 -2
  34. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-pose.yaml +23 -2
  35. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-seg.yaml +2 -2
  36. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8.yaml +2 -2
  37. ultralytics-8.3.226/ultralytics/cfg/datasets/construction-ppe.yaml +32 -0
  38. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/crack-seg.yaml +5 -5
  39. ultralytics-8.3.226/ultralytics/cfg/datasets/dog-pose.yaml +52 -0
  40. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/dota8-multispectral.yaml +2 -2
  41. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/dota8.yaml +2 -2
  42. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/hand-keypoints.yaml +29 -4
  43. ultralytics-8.3.226/ultralytics/cfg/datasets/kitti.yaml +27 -0
  44. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/lvis.yaml +9 -9
  45. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/medical-pills.yaml +4 -5
  46. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/open-images-v7.yaml +3 -3
  47. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/package-seg.yaml +5 -5
  48. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/signature.yaml +4 -4
  49. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/tiger-pose.yaml +20 -4
  50. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/xView.yaml +5 -5
  51. ultralytics-8.3.226/ultralytics/cfg/default.yaml +130 -0
  52. ultralytics-8.3.226/ultralytics/cfg/trackers/botsort.yaml +21 -0
  53. ultralytics-8.3.226/ultralytics/cfg/trackers/bytetrack.yaml +12 -0
  54. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/annotator.py +10 -11
  55. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/augment.py +499 -517
  56. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/base.py +41 -49
  57. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/build.py +180 -27
  58. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/converter.py +157 -60
  59. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/dataset.py +106 -107
  60. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/loaders.py +60 -70
  61. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/split.py +10 -10
  62. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/split_dota.py +33 -38
  63. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/utils.py +85 -84
  64. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/exporter.py +364 -491
  65. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/model.py +182 -213
  66. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/predictor.py +72 -70
  67. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/results.py +269 -341
  68. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/trainer.py +216 -120
  69. ultralytics-8.3.226/ultralytics/engine/tuner.py +448 -0
  70. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/validator.py +44 -35
  71. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/__init__.py +22 -22
  72. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/auth.py +8 -14
  73. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/google/__init__.py +17 -19
  74. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/session.py +33 -43
  75. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/utils.py +10 -115
  76. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/__init__.py +1 -1
  77. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/__init__.py +1 -1
  78. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/model.py +20 -20
  79. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/predict.py +28 -38
  80. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/utils.py +1 -2
  81. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/val.py +4 -10
  82. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/__init__.py +1 -1
  83. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/model.py +13 -13
  84. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/predict.py +7 -9
  85. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/val.py +1 -2
  86. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/__init__.py +1 -1
  87. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/model.py +7 -8
  88. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/predict.py +16 -18
  89. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/train.py +14 -16
  90. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/val.py +47 -54
  91. ultralytics-8.3.226/ultralytics/models/sam/__init__.py +12 -0
  92. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/amg.py +33 -36
  93. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/build.py +15 -17
  94. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/model.py +26 -32
  95. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/blocks.py +99 -118
  96. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/decoders.py +33 -40
  97. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/encoders.py +97 -113
  98. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/memory_attention.py +31 -36
  99. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/sam.py +69 -79
  100. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/tiny_encoder.py +103 -121
  101. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/transformer.py +46 -53
  102. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/utils.py +47 -56
  103. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/predict.py +678 -291
  104. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/loss.py +65 -75
  105. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/ops.py +28 -30
  106. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/__init__.py +1 -1
  107. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/predict.py +16 -25
  108. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/train.py +32 -67
  109. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/val.py +63 -51
  110. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/predict.py +11 -15
  111. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/train.py +55 -46
  112. ultralytics-8.3.226/ultralytics/models/yolo/detect/val.py +507 -0
  113. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/model.py +88 -96
  114. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/predict.py +5 -8
  115. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/train.py +17 -27
  116. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/val.py +94 -99
  117. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/__init__.py +1 -1
  118. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/predict.py +6 -9
  119. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/train.py +16 -62
  120. ultralytics-8.3.226/ultralytics/models/yolo/pose/val.py +259 -0
  121. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/segment/predict.py +17 -21
  122. ultralytics-8.3.226/ultralytics/models/yolo/segment/train.py +69 -0
  123. ultralytics-8.3.226/ultralytics/models/yolo/segment/val.py +249 -0
  124. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/train.py +35 -37
  125. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/train_world.py +40 -29
  126. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/predict.py +22 -29
  127. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/train.py +47 -81
  128. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/train_seg.py +5 -10
  129. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/val.py +29 -38
  130. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/__init__.py +9 -11
  131. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/autobackend.py +134 -106
  132. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/__init__.py +63 -63
  133. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/activation.py +4 -6
  134. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/block.py +145 -233
  135. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/conv.py +58 -102
  136. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/head.py +109 -155
  137. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/transformer.py +76 -105
  138. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/utils.py +15 -20
  139. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/tasks.py +132 -224
  140. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/text_model.py +68 -49
  141. ultralytics-8.3.226/ultralytics/py.typed +1 -0
  142. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/ai_gym.py +17 -23
  143. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/analytics.py +19 -18
  144. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/config.py +22 -21
  145. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/distance_calculation.py +16 -19
  146. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/heatmap.py +16 -18
  147. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/instance_segmentation.py +14 -15
  148. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_blurrer.py +13 -14
  149. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_counter.py +35 -40
  150. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_cropper.py +10 -11
  151. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/parking_management.py +35 -33
  152. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/queue_management.py +13 -13
  153. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/region_counter.py +53 -47
  154. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/security_alarm.py +22 -25
  155. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/similarity_search.py +39 -50
  156. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/solutions.py +211 -261
  157. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/speed_estimation.py +9 -11
  158. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/streamlit_inference.py +91 -32
  159. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/templates/similarity-search.html +31 -24
  160. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/trackzone.py +20 -20
  161. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/vision_eye.py +11 -12
  162. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/__init__.py +1 -1
  163. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/basetrack.py +2 -4
  164. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/bot_sort.py +27 -30
  165. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/byte_tracker.py +54 -60
  166. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/track.py +3 -8
  167. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/gmc.py +13 -18
  168. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/kalman_filter.py +35 -43
  169. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/matching.py +12 -16
  170. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/__init__.py +158 -350
  171. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/autobatch.py +5 -6
  172. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/autodevice.py +23 -25
  173. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/benchmarks.py +119 -97
  174. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/base.py +11 -12
  175. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/clearml.py +5 -6
  176. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/comet.py +90 -59
  177. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/dvc.py +13 -18
  178. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/hub.py +3 -2
  179. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/mlflow.py +4 -5
  180. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/neptune.py +4 -6
  181. ultralytics-8.3.226/ultralytics/utils/callbacks/platform.py +73 -0
  182. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/raytune.py +3 -4
  183. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/tensorboard.py +6 -8
  184. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/wb.py +19 -16
  185. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/checks.py +130 -98
  186. ultralytics-8.3.226/ultralytics/utils/cpu.py +85 -0
  187. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/dist.py +20 -16
  188. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/downloads.py +98 -98
  189. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/errors.py +6 -8
  190. ultralytics-8.3.226/ultralytics/utils/events.py +113 -0
  191. ultralytics-8.3.226/ultralytics/utils/export/__init__.py +7 -0
  192. ultralytics-8.3.148/ultralytics/utils/export.py → ultralytics-8.3.226/ultralytics/utils/export/engine.py +40 -36
  193. ultralytics-8.3.226/ultralytics/utils/export/imx.py +294 -0
  194. ultralytics-8.3.226/ultralytics/utils/export/tensorflow.py +217 -0
  195. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/files.py +19 -23
  196. ultralytics-8.3.226/ultralytics/utils/git.py +136 -0
  197. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/instance.py +68 -79
  198. ultralytics-8.3.226/ultralytics/utils/logger.py +404 -0
  199. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/loss.py +35 -34
  200. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/metrics.py +530 -519
  201. ultralytics-8.3.226/ultralytics/utils/nms.py +337 -0
  202. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/ops.py +164 -348
  203. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/patches.py +62 -21
  204. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/plotting.py +183 -198
  205. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/tal.py +28 -44
  206. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/torch_utils.py +195 -208
  207. ultralytics-8.3.226/ultralytics/utils/tqdm.py +436 -0
  208. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/triton.py +20 -21
  209. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/tuner.py +6 -5
  210. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/PKG-INFO +43 -36
  211. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/SOURCES.txt +15 -2
  212. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/requires.txt +21 -9
  213. ultralytics-8.3.148/ultralytics/__init__.py +0 -30
  214. ultralytics-8.3.148/ultralytics/cfg/datasets/VisDrone.yaml +0 -77
  215. ultralytics-8.3.148/ultralytics/cfg/datasets/dog-pose.yaml +0 -24
  216. ultralytics-8.3.148/ultralytics/cfg/default.yaml +0 -127
  217. ultralytics-8.3.148/ultralytics/cfg/trackers/botsort.yaml +0 -22
  218. ultralytics-8.3.148/ultralytics/cfg/trackers/bytetrack.yaml +0 -14
  219. ultralytics-8.3.148/ultralytics/engine/tuner.py +0 -246
  220. ultralytics-8.3.148/ultralytics/models/sam/__init__.py +0 -6
  221. ultralytics-8.3.148/ultralytics/models/yolo/detect/val.py +0 -443
  222. ultralytics-8.3.148/ultralytics/models/yolo/pose/val.py +0 -417
  223. ultralytics-8.3.148/ultralytics/models/yolo/segment/train.py +0 -127
  224. ultralytics-8.3.148/ultralytics/models/yolo/segment/val.py +0 -436
  225. {ultralytics-8.3.148 → ultralytics-8.3.226}/LICENSE +0 -0
  226. {ultralytics-8.3.148 → ultralytics-8.3.226}/setup.cfg +0 -0
  227. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/assets/bus.jpg +0 -0
  228. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/assets/zidane.jpg +0 -0
  229. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-cls-resnet18.yaml +0 -0
  230. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-cls.yaml +0 -0
  231. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-obb.yaml +0 -0
  232. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-pose.yaml +0 -0
  233. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-seg.yaml +0 -0
  234. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11.yaml +0 -0
  235. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yoloe-11-seg.yaml +0 -0
  236. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yoloe-11.yaml +0 -0
  237. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-cls.yaml +0 -0
  238. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-obb.yaml +0 -0
  239. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-pose.yaml +0 -0
  240. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-seg.yaml +0 -0
  241. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12.yaml +0 -0
  242. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml +0 -0
  243. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml +0 -0
  244. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml +0 -0
  245. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-x.yaml +0 -0
  246. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10b.yaml +0 -0
  247. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10l.yaml +0 -0
  248. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10m.yaml +0 -0
  249. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10n.yaml +0 -0
  250. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10s.yaml +0 -0
  251. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10x.yaml +0 -0
  252. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3-spp.yaml +0 -0
  253. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3-tiny.yaml +0 -0
  254. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3.yaml +0 -0
  255. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v5/yolov5-p6.yaml +0 -0
  256. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v5/yolov5.yaml +0 -0
  257. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v6/yolov6.yaml +0 -0
  258. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yoloe-v8-seg.yaml +0 -0
  259. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yoloe-v8.yaml +0 -0
  260. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls-resnet101.yaml +0 -0
  261. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls-resnet50.yaml +0 -0
  262. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls.yaml +0 -0
  263. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost-p2.yaml +0 -0
  264. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost-p6.yaml +0 -0
  265. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost.yaml +0 -0
  266. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-obb.yaml +0 -0
  267. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-p2.yaml +0 -0
  268. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-p6.yaml +0 -0
  269. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-pose-p6.yaml +0 -0
  270. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-pose.yaml +0 -0
  271. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-rtdetr.yaml +0 -0
  272. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-seg-p6.yaml +0 -0
  273. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-seg.yaml +0 -0
  274. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-world.yaml +0 -0
  275. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-worldv2.yaml +0 -0
  276. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8.yaml +0 -0
  277. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9c-seg.yaml +0 -0
  278. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9c.yaml +0 -0
  279. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9e-seg.yaml +0 -0
  280. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9e.yaml +0 -0
  281. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9m.yaml +0 -0
  282. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9s.yaml +0 -0
  283. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9t.yaml +0 -0
  284. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/__init__.py +4 -4
  285. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/download_weights.sh +0 -0
  286. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_coco.sh +0 -0
  287. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_coco128.sh +0 -0
  288. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_imagenet.sh +0 -0
  289. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/__init__.py +0 -0
  290. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/__init__.py +0 -0
  291. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/__init__.py +0 -0
  292. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/__init__.py +0 -0
  293. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/__init__.py +0 -0
  294. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/__init__.py +0 -0
  295. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/segment/__init__.py +0 -0
  296. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/__init__.py +0 -0
  297. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/__init__.py +7 -7
  298. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/__init__.py +12 -12
  299. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/__init__.py +0 -0
  300. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/__init__.py +0 -0
  301. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/dependency_links.txt +0 -0
  302. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/entry_points.txt +0 -0
  303. {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ultralytics
3
- Version: 8.3.148
3
+ Version: 8.3.226
4
4
  Summary: Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
5
5
  Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
6
6
  Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -42,11 +42,9 @@ Requires-Dist: scipy>=1.4.1
42
42
  Requires-Dist: torch>=1.8.0
43
43
  Requires-Dist: torch!=2.4.0,>=1.8.0; sys_platform == "win32"
44
44
  Requires-Dist: torchvision>=0.9.0
45
- Requires-Dist: tqdm>=4.64.0
46
45
  Requires-Dist: psutil
47
- Requires-Dist: py-cpuinfo
48
- Requires-Dist: pandas>=1.1.4
49
- Requires-Dist: ultralytics-thop>=2.0.0
46
+ Requires-Dist: polars
47
+ Requires-Dist: ultralytics-thop>=2.0.18
50
48
  Provides-Extra: dev
51
49
  Requires-Dist: ipython; extra == "dev"
52
50
  Requires-Dist: pytest; extra == "dev"
@@ -55,21 +53,23 @@ Requires-Dist: coverage[toml]; extra == "dev"
55
53
  Requires-Dist: mkdocs>=1.6.0; extra == "dev"
56
54
  Requires-Dist: mkdocs-material>=9.5.9; extra == "dev"
57
55
  Requires-Dist: mkdocstrings[python]; extra == "dev"
58
- Requires-Dist: mkdocs-ultralytics-plugin>=0.1.19; extra == "dev"
56
+ Requires-Dist: mkdocs-ultralytics-plugin>=0.1.29; extra == "dev"
59
57
  Requires-Dist: mkdocs-macros-plugin>=1.0.5; extra == "dev"
60
58
  Provides-Extra: export
61
- Requires-Dist: onnx<1.18.0,>=1.12.0; extra == "export"
59
+ Requires-Dist: numpy<2.0.0; extra == "export"
60
+ Requires-Dist: onnx>=1.12.0; platform_system != "Darwin" and extra == "export"
61
+ Requires-Dist: onnx<1.18.0,>=1.12.0; platform_system == "Darwin" and extra == "export"
62
62
  Requires-Dist: coremltools>=8.0; (platform_system != "Windows" and python_version <= "3.13") and extra == "export"
63
63
  Requires-Dist: scikit-learn>=1.3.2; (platform_system != "Windows" and python_version <= "3.13") and extra == "export"
64
64
  Requires-Dist: openvino>=2024.0.0; extra == "export"
65
- Requires-Dist: tensorflow>=2.0.0; extra == "export"
65
+ Requires-Dist: tensorflow<=2.19.0,>=2.0.0; extra == "export"
66
66
  Requires-Dist: tensorflowjs>=2.0.0; extra == "export"
67
67
  Requires-Dist: tensorstore>=0.1.63; (platform_machine == "aarch64" and python_version >= "3.9") and extra == "export"
68
68
  Requires-Dist: h5py!=3.11.0; platform_machine == "aarch64" and extra == "export"
69
69
  Provides-Extra: solutions
70
70
  Requires-Dist: shapely>=2.0.0; extra == "solutions"
71
71
  Requires-Dist: streamlit>=1.29.0; extra == "solutions"
72
- Requires-Dist: flask; extra == "solutions"
72
+ Requires-Dist: flask>=3.0.1; extra == "solutions"
73
73
  Provides-Extra: logging
74
74
  Requires-Dist: wandb; extra == "logging"
75
75
  Requires-Dist: tensorboard; extra == "logging"
@@ -78,20 +78,27 @@ Provides-Extra: extra
78
78
  Requires-Dist: hub-sdk>=0.0.12; extra == "extra"
79
79
  Requires-Dist: ipython; extra == "extra"
80
80
  Requires-Dist: albumentations>=1.4.6; extra == "extra"
81
- Requires-Dist: pycocotools>=2.0.7; extra == "extra"
81
+ Requires-Dist: faster-coco-eval>=1.6.7; extra == "extra"
82
+ Provides-Extra: typing
83
+ Requires-Dist: scipy-stubs; extra == "typing"
84
+ Requires-Dist: types-pillow; extra == "typing"
85
+ Requires-Dist: types-psutil; extra == "typing"
86
+ Requires-Dist: types-pyyaml; extra == "typing"
87
+ Requires-Dist: types-requests; extra == "typing"
88
+ Requires-Dist: types-shapely; extra == "typing"
82
89
  Dynamic: license-file
83
90
 
84
91
  <div align="center">
85
92
  <p>
86
- <a href="https://www.ultralytics.com/blog/ultralytics-yolo11-has-arrived-redefine-whats-possible-in-ai" target="_blank">
93
+ <a href="https://www.ultralytics.com/events/yolovision?utm_source=github&utm_medium=org&utm_campaign=yv25_event" target="_blank">
87
94
  <img width="100%" src="https://raw.githubusercontent.com/ultralytics/assets/main/yolov8/banner-yolov8.png" alt="Ultralytics YOLO banner"></a>
88
95
  </p>
89
96
 
90
- [中文](https://docs.ultralytics.com/zh) | [한국어](https://docs.ultralytics.com/ko) | [日本語](https://docs.ultralytics.com/ja) | [Русский](https://docs.ultralytics.com/ru) | [Deutsch](https://docs.ultralytics.com/de) | [Français](https://docs.ultralytics.com/fr) | [Español](https://docs.ultralytics.com/es) | [Português](https://docs.ultralytics.com/pt) | [Türkçe](https://docs.ultralytics.com/tr) | [Tiếng Việt](https://docs.ultralytics.com/vi) | [العربية](https://docs.ultralytics.com/ar) <br>
97
+ [中文](https://docs.ultralytics.com/zh/) | [한국어](https://docs.ultralytics.com/ko/) | [日本語](https://docs.ultralytics.com/ja/) | [Русский](https://docs.ultralytics.com/ru/) | [Deutsch](https://docs.ultralytics.com/de/) | [Français](https://docs.ultralytics.com/fr/) | [Español](https://docs.ultralytics.com/es) | [Português](https://docs.ultralytics.com/pt/) | [Türkçe](https://docs.ultralytics.com/tr/) | [Tiếng Việt](https://docs.ultralytics.com/vi/) | [العربية](https://docs.ultralytics.com/ar/) <br>
91
98
 
92
99
  <div>
93
100
  <a href="https://github.com/ultralytics/ultralytics/actions/workflows/ci.yml"><img src="https://github.com/ultralytics/ultralytics/actions/workflows/ci.yml/badge.svg" alt="Ultralytics CI"></a>
94
- <a href="https://pepy.tech/projects/ultralytics"><img src="https://static.pepy.tech/badge/ultralytics" alt="Ultralytics Downloads"></a>
101
+ <a href="https://clickpy.clickhouse.com/dashboard/ultralytics"><img src="https://static.pepy.tech/badge/ultralytics" alt="Ultralytics Downloads"></a>
95
102
  <a href="https://zenodo.org/badge/latestdoi/264818686"><img src="https://zenodo.org/badge/264818686.svg" alt="Ultralytics YOLO Citation"></a>
96
103
  <a href="https://discord.com/invite/ultralytics"><img alt="Ultralytics Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
97
104
  <a href="https://community.ultralytics.com/"><img alt="Ultralytics Forums" src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue"></a>
@@ -122,7 +129,7 @@ Request an Enterprise License for commercial use at [Ultralytics Licensing](http
122
129
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
123
130
  <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="2%" alt="Ultralytics Twitter"></a>
124
131
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
125
- <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="2%" alt="Ultralytics YouTube"></a>
132
+ <a href="https://www.youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="2%" alt="Ultralytics YouTube"></a>
126
133
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
127
134
  <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="2%" alt="Ultralytics TikTok"></a>
128
135
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
@@ -140,7 +147,7 @@ See below for quickstart installation and usage examples. For comprehensive guid
140
147
 
141
148
  Install the `ultralytics` package, including all [requirements](https://github.com/ultralytics/ultralytics/blob/main/pyproject.toml), in a [**Python>=3.8**](https://www.python.org/) environment with [**PyTorch>=1.8**](https://pytorch.org/get-started/locally/).
142
149
 
143
- [![PyPI - Version](https://img.shields.io/pypi/v/ultralytics?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics/) [![Ultralytics Downloads](https://static.pepy.tech/badge/ultralytics)](https://www.pepy.tech/projects/ultralytics) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics/)
150
+ [![PyPI - Version](https://img.shields.io/pypi/v/ultralytics?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics/) [![Ultralytics Downloads](https://static.pepy.tech/badge/ultralytics)](https://clickpy.clickhouse.com/dashboard/ultralytics) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics/)
144
151
 
145
152
  ```bash
146
153
  pip install ultralytics
@@ -232,11 +239,11 @@ Refer to the [Segmentation Docs](https://docs.ultralytics.com/tasks/segment/) fo
232
239
 
233
240
  | Model | size<br><sup>(pixels) | mAP<sup>box<br>50-95 | mAP<sup>mask<br>50-95 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
234
241
  | -------------------------------------------------------------------------------------------- | --------------------- | -------------------- | --------------------- | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
235
- | [YOLO11n-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.pt) | 640 | 38.9 | 32.0 | 65.9 ± 1.1 | 1.8 ± 0.0 | 2.9 | 10.4 |
236
- | [YOLO11s-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-seg.pt) | 640 | 46.6 | 37.8 | 117.6 ± 4.9 | 2.9 ± 0.0 | 10.1 | 35.5 |
237
- | [YOLO11m-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-seg.pt) | 640 | 51.5 | 41.5 | 281.6 ± 1.2 | 6.3 ± 0.1 | 22.4 | 123.3 |
238
- | [YOLO11l-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-seg.pt) | 640 | 53.4 | 42.9 | 344.2 ± 3.2 | 7.8 ± 0.2 | 27.6 | 142.2 |
239
- | [YOLO11x-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-seg.pt) | 640 | 54.7 | 43.8 | 664.5 ± 3.2 | 15.8 ± 0.7 | 62.1 | 319.0 |
242
+ | [YOLO11n-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.pt) | 640 | 38.9 | 32.0 | 65.9 ± 1.1 | 1.8 ± 0.0 | 2.9 | 9.7 |
243
+ | [YOLO11s-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-seg.pt) | 640 | 46.6 | 37.8 | 117.6 ± 4.9 | 2.9 ± 0.0 | 10.1 | 33.0 |
244
+ | [YOLO11m-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-seg.pt) | 640 | 51.5 | 41.5 | 281.6 ± 1.2 | 6.3 ± 0.1 | 22.4 | 113.2 |
245
+ | [YOLO11l-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-seg.pt) | 640 | 53.4 | 42.9 | 344.2 ± 3.2 | 7.8 ± 0.2 | 27.6 | 132.2 |
246
+ | [YOLO11x-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-seg.pt) | 640 | 54.7 | 43.8 | 664.5 ± 3.2 | 15.8 ± 0.7 | 62.1 | 296.4 |
240
247
 
241
248
  - **mAP<sup>val</sup>** values are for single-model single-scale on the [COCO val2017](https://cocodataset.org/) dataset. See [YOLO Performance Metrics](https://docs.ultralytics.com/guides/yolo-performance-metrics/) for details. <br>Reproduce with `yolo val segment data=coco.yaml device=0`
242
249
  - **Speed** metrics are averaged over COCO val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val segment data=coco.yaml batch=1 device=0|cpu`
@@ -249,11 +256,11 @@ Consult the [Classification Docs](https://docs.ultralytics.com/tasks/classify/)
249
256
 
250
257
  | Model | size<br><sup>(pixels) | acc<br><sup>top1 | acc<br><sup>top5 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) at 224 |
251
258
  | -------------------------------------------------------------------------------------------- | --------------------- | ---------------- | ---------------- | ------------------------------ | ----------------------------------- | ------------------ | ------------------------ |
252
- | [YOLO11n-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.pt) | 224 | 70.0 | 89.4 | 5.0 ± 0.3 | 1.1 ± 0.0 | 1.6 | 0.5 |
253
- | [YOLO11s-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-cls.pt) | 224 | 75.4 | 92.7 | 7.9 ± 0.2 | 1.3 ± 0.0 | 5.5 | 1.6 |
254
- | [YOLO11m-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-cls.pt) | 224 | 77.3 | 93.9 | 17.2 ± 0.4 | 2.0 ± 0.0 | 10.4 | 5.0 |
255
- | [YOLO11l-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-cls.pt) | 224 | 78.3 | 94.3 | 23.2 ± 0.3 | 2.8 ± 0.0 | 12.9 | 6.2 |
256
- | [YOLO11x-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-cls.pt) | 224 | 79.5 | 94.9 | 41.4 ± 0.9 | 3.8 ± 0.0 | 28.4 | 13.7 |
259
+ | [YOLO11n-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.pt) | 224 | 70.0 | 89.4 | 5.0 ± 0.3 | 1.1 ± 0.0 | 2.8 | 0.5 |
260
+ | [YOLO11s-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-cls.pt) | 224 | 75.4 | 92.7 | 7.9 ± 0.2 | 1.3 ± 0.0 | 6.7 | 1.6 |
261
+ | [YOLO11m-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-cls.pt) | 224 | 77.3 | 93.9 | 17.2 ± 0.4 | 2.0 ± 0.0 | 11.6 | 4.9 |
262
+ | [YOLO11l-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-cls.pt) | 224 | 78.3 | 94.3 | 23.2 ± 0.3 | 2.8 ± 0.0 | 14.1 | 6.2 |
263
+ | [YOLO11x-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-cls.pt) | 224 | 79.5 | 94.9 | 41.4 ± 0.9 | 3.8 ± 0.0 | 29.6 | 13.6 |
257
264
 
258
265
  - **acc** values represent model accuracy on the [ImageNet](https://www.image-net.org/) dataset validation set. <br>Reproduce with `yolo val classify data=path/to/ImageNet device=0`
259
266
  - **Speed** metrics are averaged over ImageNet val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val classify data=path/to/ImageNet batch=1 device=0|cpu`
@@ -266,11 +273,11 @@ See the [Pose Estimation Docs](https://docs.ultralytics.com/tasks/pose/) for usa
266
273
 
267
274
  | Model | size<br><sup>(pixels) | mAP<sup>pose<br>50-95 | mAP<sup>pose<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
268
275
  | ---------------------------------------------------------------------------------------------- | --------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
269
- | [YOLO11n-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.pt) | 640 | 50.0 | 81.0 | 52.4 ± 0.5 | 1.7 ± 0.0 | 2.9 | 7.6 |
270
- | [YOLO11s-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-pose.pt) | 640 | 58.9 | 86.3 | 90.5 ± 0.6 | 2.6 ± 0.0 | 9.9 | 23.2 |
271
- | [YOLO11m-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-pose.pt) | 640 | 64.9 | 89.4 | 187.3 ± 0.8 | 4.9 ± 0.1 | 20.9 | 71.7 |
272
- | [YOLO11l-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-pose.pt) | 640 | 66.1 | 89.9 | 247.7 ± 1.1 | 6.4 ± 0.1 | 26.2 | 90.7 |
273
- | [YOLO11x-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-pose.pt) | 640 | 69.5 | 91.1 | 488.0 ± 13.9 | 12.1 ± 0.2 | 58.8 | 203.3 |
276
+ | [YOLO11n-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.pt) | 640 | 50.0 | 81.0 | 52.4 ± 0.5 | 1.7 ± 0.0 | 2.9 | 7.4 |
277
+ | [YOLO11s-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-pose.pt) | 640 | 58.9 | 86.3 | 90.5 ± 0.6 | 2.6 ± 0.0 | 9.9 | 23.1 |
278
+ | [YOLO11m-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-pose.pt) | 640 | 64.9 | 89.4 | 187.3 ± 0.8 | 4.9 ± 0.1 | 20.9 | 71.4 |
279
+ | [YOLO11l-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-pose.pt) | 640 | 66.1 | 89.9 | 247.7 ± 1.1 | 6.4 ± 0.1 | 26.1 | 90.3 |
280
+ | [YOLO11x-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-pose.pt) | 640 | 69.5 | 91.1 | 488.0 ± 13.9 | 12.1 ± 0.2 | 58.8 | 202.8 |
274
281
 
275
282
  - **mAP<sup>val</sup>** values are for single-model single-scale on the [COCO Keypoints val2017](https://docs.ultralytics.com/datasets/pose/coco/) dataset. See [YOLO Performance Metrics](https://docs.ultralytics.com/guides/yolo-performance-metrics/) for details. <br>Reproduce with `yolo val pose data=coco-pose.yaml device=0`
276
283
  - **Speed** metrics are averaged over COCO val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val pose data=coco-pose.yaml batch=1 device=0|cpu`
@@ -283,11 +290,11 @@ Check the [OBB Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples
283
290
 
284
291
  | Model | size<br><sup>(pixels) | mAP<sup>test<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
285
292
  | -------------------------------------------------------------------------------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
286
- | [YOLO11n-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.pt) | 1024 | 78.4 | 117.6 ± 0.8 | 4.4 ± 0.0 | 2.7 | 17.2 |
287
- | [YOLO11s-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-obb.pt) | 1024 | 79.5 | 219.4 ± 4.0 | 5.1 ± 0.0 | 9.7 | 57.5 |
288
- | [YOLO11m-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-obb.pt) | 1024 | 80.9 | 562.8 ± 2.9 | 10.1 ± 0.4 | 20.9 | 183.5 |
289
- | [YOLO11l-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-obb.pt) | 1024 | 81.0 | 712.5 ± 5.0 | 13.5 ± 0.6 | 26.2 | 232.0 |
290
- | [YOLO11x-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-obb.pt) | 1024 | 81.3 | 1408.6 ± 7.7 | 28.6 ± 1.0 | 58.8 | 520.2 |
293
+ | [YOLO11n-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.pt) | 1024 | 78.4 | 117.6 ± 0.8 | 4.4 ± 0.0 | 2.7 | 16.8 |
294
+ | [YOLO11s-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-obb.pt) | 1024 | 79.5 | 219.4 ± 4.0 | 5.1 ± 0.0 | 9.7 | 57.1 |
295
+ | [YOLO11m-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-obb.pt) | 1024 | 80.9 | 562.8 ± 2.9 | 10.1 ± 0.4 | 20.9 | 182.8 |
296
+ | [YOLO11l-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-obb.pt) | 1024 | 81.0 | 712.5 ± 5.0 | 13.5 ± 0.6 | 26.1 | 231.2 |
297
+ | [YOLO11x-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-obb.pt) | 1024 | 81.3 | 1408.6 ± 7.7 | 28.6 ± 1.0 | 58.8 | 519.1 |
291
298
 
292
299
  - **mAP<sup>test</sup>** values are for single-model multiscale performance on the [DOTAv1 test set](https://captain-whu.github.io/DOTA/dataset.html). <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test` and submit merged results to the [DOTA evaluation server](https://captain-whu.github.io/DOTA/evaluation.html).
293
300
  - **Speed** metrics are averaged over [DOTAv1 val images](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10) using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce by `yolo val obb data=DOTAv1.yaml batch=1 device=0|cpu`
@@ -358,7 +365,7 @@ For bug reports and feature requests related to Ultralytics software, please vis
358
365
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
359
366
  <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="Ultralytics Twitter"></a>
360
367
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
361
- <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
368
+ <a href="https://www.youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
362
369
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
363
370
  <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
364
371
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
@@ -1,14 +1,14 @@
1
1
  <div align="center">
2
2
  <p>
3
- <a href="https://www.ultralytics.com/blog/ultralytics-yolo11-has-arrived-redefine-whats-possible-in-ai" target="_blank">
3
+ <a href="https://www.ultralytics.com/events/yolovision?utm_source=github&utm_medium=org&utm_campaign=yv25_event" target="_blank">
4
4
  <img width="100%" src="https://raw.githubusercontent.com/ultralytics/assets/main/yolov8/banner-yolov8.png" alt="Ultralytics YOLO banner"></a>
5
5
  </p>
6
6
 
7
- [中文](https://docs.ultralytics.com/zh) | [한국어](https://docs.ultralytics.com/ko) | [日本語](https://docs.ultralytics.com/ja) | [Русский](https://docs.ultralytics.com/ru) | [Deutsch](https://docs.ultralytics.com/de) | [Français](https://docs.ultralytics.com/fr) | [Español](https://docs.ultralytics.com/es) | [Português](https://docs.ultralytics.com/pt) | [Türkçe](https://docs.ultralytics.com/tr) | [Tiếng Việt](https://docs.ultralytics.com/vi) | [العربية](https://docs.ultralytics.com/ar) <br>
7
+ [中文](https://docs.ultralytics.com/zh/) | [한국어](https://docs.ultralytics.com/ko/) | [日本語](https://docs.ultralytics.com/ja/) | [Русский](https://docs.ultralytics.com/ru/) | [Deutsch](https://docs.ultralytics.com/de/) | [Français](https://docs.ultralytics.com/fr/) | [Español](https://docs.ultralytics.com/es) | [Português](https://docs.ultralytics.com/pt/) | [Türkçe](https://docs.ultralytics.com/tr/) | [Tiếng Việt](https://docs.ultralytics.com/vi/) | [العربية](https://docs.ultralytics.com/ar/) <br>
8
8
 
9
9
  <div>
10
10
  <a href="https://github.com/ultralytics/ultralytics/actions/workflows/ci.yml"><img src="https://github.com/ultralytics/ultralytics/actions/workflows/ci.yml/badge.svg" alt="Ultralytics CI"></a>
11
- <a href="https://pepy.tech/projects/ultralytics"><img src="https://static.pepy.tech/badge/ultralytics" alt="Ultralytics Downloads"></a>
11
+ <a href="https://clickpy.clickhouse.com/dashboard/ultralytics"><img src="https://static.pepy.tech/badge/ultralytics" alt="Ultralytics Downloads"></a>
12
12
  <a href="https://zenodo.org/badge/latestdoi/264818686"><img src="https://zenodo.org/badge/264818686.svg" alt="Ultralytics YOLO Citation"></a>
13
13
  <a href="https://discord.com/invite/ultralytics"><img alt="Ultralytics Discord" src="https://img.shields.io/discord/1089800235347353640?logo=discord&logoColor=white&label=Discord&color=blue"></a>
14
14
  <a href="https://community.ultralytics.com/"><img alt="Ultralytics Forums" src="https://img.shields.io/discourse/users?server=https%3A%2F%2Fcommunity.ultralytics.com&logo=discourse&label=Forums&color=blue"></a>
@@ -39,7 +39,7 @@ Request an Enterprise License for commercial use at [Ultralytics Licensing](http
39
39
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
40
40
  <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="2%" alt="Ultralytics Twitter"></a>
41
41
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
42
- <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="2%" alt="Ultralytics YouTube"></a>
42
+ <a href="https://www.youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="2%" alt="Ultralytics YouTube"></a>
43
43
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
44
44
  <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="2%" alt="Ultralytics TikTok"></a>
45
45
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="2%" alt="space">
@@ -57,7 +57,7 @@ See below for quickstart installation and usage examples. For comprehensive guid
57
57
 
58
58
  Install the `ultralytics` package, including all [requirements](https://github.com/ultralytics/ultralytics/blob/main/pyproject.toml), in a [**Python>=3.8**](https://www.python.org/) environment with [**PyTorch>=1.8**](https://pytorch.org/get-started/locally/).
59
59
 
60
- [![PyPI - Version](https://img.shields.io/pypi/v/ultralytics?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics/) [![Ultralytics Downloads](https://static.pepy.tech/badge/ultralytics)](https://www.pepy.tech/projects/ultralytics) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics/)
60
+ [![PyPI - Version](https://img.shields.io/pypi/v/ultralytics?logo=pypi&logoColor=white)](https://pypi.org/project/ultralytics/) [![Ultralytics Downloads](https://static.pepy.tech/badge/ultralytics)](https://clickpy.clickhouse.com/dashboard/ultralytics) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/ultralytics?logo=python&logoColor=gold)](https://pypi.org/project/ultralytics/)
61
61
 
62
62
  ```bash
63
63
  pip install ultralytics
@@ -149,11 +149,11 @@ Refer to the [Segmentation Docs](https://docs.ultralytics.com/tasks/segment/) fo
149
149
 
150
150
  | Model | size<br><sup>(pixels) | mAP<sup>box<br>50-95 | mAP<sup>mask<br>50-95 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
151
151
  | -------------------------------------------------------------------------------------------- | --------------------- | -------------------- | --------------------- | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
152
- | [YOLO11n-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.pt) | 640 | 38.9 | 32.0 | 65.9 ± 1.1 | 1.8 ± 0.0 | 2.9 | 10.4 |
153
- | [YOLO11s-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-seg.pt) | 640 | 46.6 | 37.8 | 117.6 ± 4.9 | 2.9 ± 0.0 | 10.1 | 35.5 |
154
- | [YOLO11m-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-seg.pt) | 640 | 51.5 | 41.5 | 281.6 ± 1.2 | 6.3 ± 0.1 | 22.4 | 123.3 |
155
- | [YOLO11l-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-seg.pt) | 640 | 53.4 | 42.9 | 344.2 ± 3.2 | 7.8 ± 0.2 | 27.6 | 142.2 |
156
- | [YOLO11x-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-seg.pt) | 640 | 54.7 | 43.8 | 664.5 ± 3.2 | 15.8 ± 0.7 | 62.1 | 319.0 |
152
+ | [YOLO11n-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.pt) | 640 | 38.9 | 32.0 | 65.9 ± 1.1 | 1.8 ± 0.0 | 2.9 | 9.7 |
153
+ | [YOLO11s-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-seg.pt) | 640 | 46.6 | 37.8 | 117.6 ± 4.9 | 2.9 ± 0.0 | 10.1 | 33.0 |
154
+ | [YOLO11m-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-seg.pt) | 640 | 51.5 | 41.5 | 281.6 ± 1.2 | 6.3 ± 0.1 | 22.4 | 113.2 |
155
+ | [YOLO11l-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-seg.pt) | 640 | 53.4 | 42.9 | 344.2 ± 3.2 | 7.8 ± 0.2 | 27.6 | 132.2 |
156
+ | [YOLO11x-seg](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-seg.pt) | 640 | 54.7 | 43.8 | 664.5 ± 3.2 | 15.8 ± 0.7 | 62.1 | 296.4 |
157
157
 
158
158
  - **mAP<sup>val</sup>** values are for single-model single-scale on the [COCO val2017](https://cocodataset.org/) dataset. See [YOLO Performance Metrics](https://docs.ultralytics.com/guides/yolo-performance-metrics/) for details. <br>Reproduce with `yolo val segment data=coco.yaml device=0`
159
159
  - **Speed** metrics are averaged over COCO val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val segment data=coco.yaml batch=1 device=0|cpu`
@@ -166,11 +166,11 @@ Consult the [Classification Docs](https://docs.ultralytics.com/tasks/classify/)
166
166
 
167
167
  | Model | size<br><sup>(pixels) | acc<br><sup>top1 | acc<br><sup>top5 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) at 224 |
168
168
  | -------------------------------------------------------------------------------------------- | --------------------- | ---------------- | ---------------- | ------------------------------ | ----------------------------------- | ------------------ | ------------------------ |
169
- | [YOLO11n-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.pt) | 224 | 70.0 | 89.4 | 5.0 ± 0.3 | 1.1 ± 0.0 | 1.6 | 0.5 |
170
- | [YOLO11s-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-cls.pt) | 224 | 75.4 | 92.7 | 7.9 ± 0.2 | 1.3 ± 0.0 | 5.5 | 1.6 |
171
- | [YOLO11m-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-cls.pt) | 224 | 77.3 | 93.9 | 17.2 ± 0.4 | 2.0 ± 0.0 | 10.4 | 5.0 |
172
- | [YOLO11l-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-cls.pt) | 224 | 78.3 | 94.3 | 23.2 ± 0.3 | 2.8 ± 0.0 | 12.9 | 6.2 |
173
- | [YOLO11x-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-cls.pt) | 224 | 79.5 | 94.9 | 41.4 ± 0.9 | 3.8 ± 0.0 | 28.4 | 13.7 |
169
+ | [YOLO11n-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.pt) | 224 | 70.0 | 89.4 | 5.0 ± 0.3 | 1.1 ± 0.0 | 2.8 | 0.5 |
170
+ | [YOLO11s-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-cls.pt) | 224 | 75.4 | 92.7 | 7.9 ± 0.2 | 1.3 ± 0.0 | 6.7 | 1.6 |
171
+ | [YOLO11m-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-cls.pt) | 224 | 77.3 | 93.9 | 17.2 ± 0.4 | 2.0 ± 0.0 | 11.6 | 4.9 |
172
+ | [YOLO11l-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-cls.pt) | 224 | 78.3 | 94.3 | 23.2 ± 0.3 | 2.8 ± 0.0 | 14.1 | 6.2 |
173
+ | [YOLO11x-cls](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-cls.pt) | 224 | 79.5 | 94.9 | 41.4 ± 0.9 | 3.8 ± 0.0 | 29.6 | 13.6 |
174
174
 
175
175
  - **acc** values represent model accuracy on the [ImageNet](https://www.image-net.org/) dataset validation set. <br>Reproduce with `yolo val classify data=path/to/ImageNet device=0`
176
176
  - **Speed** metrics are averaged over ImageNet val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val classify data=path/to/ImageNet batch=1 device=0|cpu`
@@ -183,11 +183,11 @@ See the [Pose Estimation Docs](https://docs.ultralytics.com/tasks/pose/) for usa
183
183
 
184
184
  | Model | size<br><sup>(pixels) | mAP<sup>pose<br>50-95 | mAP<sup>pose<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
185
185
  | ---------------------------------------------------------------------------------------------- | --------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
186
- | [YOLO11n-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.pt) | 640 | 50.0 | 81.0 | 52.4 ± 0.5 | 1.7 ± 0.0 | 2.9 | 7.6 |
187
- | [YOLO11s-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-pose.pt) | 640 | 58.9 | 86.3 | 90.5 ± 0.6 | 2.6 ± 0.0 | 9.9 | 23.2 |
188
- | [YOLO11m-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-pose.pt) | 640 | 64.9 | 89.4 | 187.3 ± 0.8 | 4.9 ± 0.1 | 20.9 | 71.7 |
189
- | [YOLO11l-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-pose.pt) | 640 | 66.1 | 89.9 | 247.7 ± 1.1 | 6.4 ± 0.1 | 26.2 | 90.7 |
190
- | [YOLO11x-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-pose.pt) | 640 | 69.5 | 91.1 | 488.0 ± 13.9 | 12.1 ± 0.2 | 58.8 | 203.3 |
186
+ | [YOLO11n-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.pt) | 640 | 50.0 | 81.0 | 52.4 ± 0.5 | 1.7 ± 0.0 | 2.9 | 7.4 |
187
+ | [YOLO11s-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-pose.pt) | 640 | 58.9 | 86.3 | 90.5 ± 0.6 | 2.6 ± 0.0 | 9.9 | 23.1 |
188
+ | [YOLO11m-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-pose.pt) | 640 | 64.9 | 89.4 | 187.3 ± 0.8 | 4.9 ± 0.1 | 20.9 | 71.4 |
189
+ | [YOLO11l-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-pose.pt) | 640 | 66.1 | 89.9 | 247.7 ± 1.1 | 6.4 ± 0.1 | 26.1 | 90.3 |
190
+ | [YOLO11x-pose](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-pose.pt) | 640 | 69.5 | 91.1 | 488.0 ± 13.9 | 12.1 ± 0.2 | 58.8 | 202.8 |
191
191
 
192
192
  - **mAP<sup>val</sup>** values are for single-model single-scale on the [COCO Keypoints val2017](https://docs.ultralytics.com/datasets/pose/coco/) dataset. See [YOLO Performance Metrics](https://docs.ultralytics.com/guides/yolo-performance-metrics/) for details. <br>Reproduce with `yolo val pose data=coco-pose.yaml device=0`
193
193
  - **Speed** metrics are averaged over COCO val images using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce with `yolo val pose data=coco-pose.yaml batch=1 device=0|cpu`
@@ -200,11 +200,11 @@ Check the [OBB Docs](https://docs.ultralytics.com/tasks/obb/) for usage examples
200
200
 
201
201
  | Model | size<br><sup>(pixels) | mAP<sup>test<br>50 | Speed<br><sup>CPU ONNX<br>(ms) | Speed<br><sup>T4 TensorRT10<br>(ms) | params<br><sup>(M) | FLOPs<br><sup>(B) |
202
202
  | -------------------------------------------------------------------------------------------- | --------------------- | ------------------ | ------------------------------ | ----------------------------------- | ------------------ | ----------------- |
203
- | [YOLO11n-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.pt) | 1024 | 78.4 | 117.6 ± 0.8 | 4.4 ± 0.0 | 2.7 | 17.2 |
204
- | [YOLO11s-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-obb.pt) | 1024 | 79.5 | 219.4 ± 4.0 | 5.1 ± 0.0 | 9.7 | 57.5 |
205
- | [YOLO11m-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-obb.pt) | 1024 | 80.9 | 562.8 ± 2.9 | 10.1 ± 0.4 | 20.9 | 183.5 |
206
- | [YOLO11l-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-obb.pt) | 1024 | 81.0 | 712.5 ± 5.0 | 13.5 ± 0.6 | 26.2 | 232.0 |
207
- | [YOLO11x-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-obb.pt) | 1024 | 81.3 | 1408.6 ± 7.7 | 28.6 ± 1.0 | 58.8 | 520.2 |
203
+ | [YOLO11n-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.pt) | 1024 | 78.4 | 117.6 ± 0.8 | 4.4 ± 0.0 | 2.7 | 16.8 |
204
+ | [YOLO11s-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s-obb.pt) | 1024 | 79.5 | 219.4 ± 4.0 | 5.1 ± 0.0 | 9.7 | 57.1 |
205
+ | [YOLO11m-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m-obb.pt) | 1024 | 80.9 | 562.8 ± 2.9 | 10.1 ± 0.4 | 20.9 | 182.8 |
206
+ | [YOLO11l-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l-obb.pt) | 1024 | 81.0 | 712.5 ± 5.0 | 13.5 ± 0.6 | 26.1 | 231.2 |
207
+ | [YOLO11x-obb](https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x-obb.pt) | 1024 | 81.3 | 1408.6 ± 7.7 | 28.6 ± 1.0 | 58.8 | 519.1 |
208
208
 
209
209
  - **mAP<sup>test</sup>** values are for single-model multiscale performance on the [DOTAv1 test set](https://captain-whu.github.io/DOTA/dataset.html). <br>Reproduce by `yolo val obb data=DOTAv1.yaml device=0 split=test` and submit merged results to the [DOTA evaluation server](https://captain-whu.github.io/DOTA/evaluation.html).
210
210
  - **Speed** metrics are averaged over [DOTAv1 val images](https://docs.ultralytics.com/datasets/obb/dota-v2/#dota-v10) using an [Amazon EC2 P4d](https://aws.amazon.com/ec2/instance-types/p4/) instance. CPU speeds measured with [ONNX](https://onnx.ai/) export. GPU speeds measured with [TensorRT](https://developer.nvidia.com/tensorrt) export. <br>Reproduce by `yolo val obb data=DOTAv1.yaml batch=1 device=0|cpu`
@@ -275,7 +275,7 @@ For bug reports and feature requests related to Ultralytics software, please vis
275
275
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
276
276
  <a href="https://twitter.com/ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-twitter.png" width="3%" alt="Ultralytics Twitter"></a>
277
277
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
278
- <a href="https://youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
278
+ <a href="https://www.youtube.com/ultralytics?sub_confirmation=1"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-youtube.png" width="3%" alt="Ultralytics YouTube"></a>
279
279
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
280
280
  <a href="https://www.tiktok.com/@ultralytics"><img src="https://github.com/ultralytics/assets/raw/main/social/logo-social-tiktok.png" width="3%" alt="Ultralytics TikTok"></a>
281
281
  <img src="https://github.com/ultralytics/assets/raw/main/social/logo-transparent.png" width="3%" alt="space">
@@ -71,11 +71,9 @@ dependencies = [
71
71
  "torch>=1.8.0",
72
72
  "torch>=1.8.0,!=2.4.0; sys_platform == 'win32'", # Windows CPU errors w/ 2.4.0 https://github.com/ultralytics/ultralytics/issues/15049
73
73
  "torchvision>=0.9.0",
74
- "tqdm>=4.64.0", # progress bars
75
74
  "psutil", # system utilization
76
- "py-cpuinfo", # display CPU info
77
- "pandas>=1.1.4",
78
- "ultralytics-thop>=2.0.0", # FLOPs computation https://github.com/ultralytics/thop
75
+ "polars",
76
+ "ultralytics-thop>=2.0.18", # FLOPs computation https://github.com/ultralytics/thop
79
77
  ]
80
78
 
81
79
  # Optional dependencies ------------------------------------------------------------------------------------------------
@@ -88,15 +86,17 @@ dev = [
88
86
  "mkdocs>=1.6.0",
89
87
  "mkdocs-material>=9.5.9",
90
88
  "mkdocstrings[python]",
91
- "mkdocs-ultralytics-plugin>=0.1.19", # for meta descriptions and images, dates and authors
92
- "mkdocs-macros-plugin>=1.0.5" # duplicating content (i.e. export tables) in multiple places
89
+ "mkdocs-ultralytics-plugin>=0.1.29", # for meta descriptions and images, dates and authors
90
+ "mkdocs-macros-plugin>=1.0.5", # duplicating content (i.e. export tables) in multiple places
93
91
  ]
94
92
  export = [
95
- "onnx>=1.12.0,<1.18.0", # ONNX export
93
+ "numpy<2.0.0", # TF 2.20 compatibility
94
+ "onnx>=1.12.0; platform_system != 'Darwin'", # ONNX export
95
+ "onnx>=1.12.0,<1.18.0; platform_system == 'Darwin'", # TF inference hanging on MacOS
96
96
  "coremltools>=8.0; platform_system != 'Windows' and python_version <= '3.13'", # CoreML supported on macOS and Linux
97
97
  "scikit-learn>=1.3.2; platform_system != 'Windows' and python_version <= '3.13'", # CoreML k-means quantization
98
98
  "openvino>=2024.0.0", # OpenVINO export
99
- "tensorflow>=2.0.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
99
+ "tensorflow>=2.0.0,<=2.19.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
100
100
  "tensorflowjs>=2.0.0", # TF.js export, automatically installs tensorflow
101
101
  "tensorstore>=0.1.63; platform_machine == 'aarch64' and python_version >= '3.9'", # for TF Raspberry Pi exports
102
102
  "h5py!=3.11.0; platform_machine == 'aarch64'", # fix h5py build issues due to missing aarch64 wheels in 3.11 release
@@ -104,7 +104,7 @@ export = [
104
104
  solutions = [
105
105
  "shapely>=2.0.0", # shapely for point and polygon data matching
106
106
  "streamlit>=1.29.0", # for live inference on web browser, i.e `yolo streamlit-predict`
107
- "flask", # for similarity search solution
107
+ "flask>=3.0.1", # for similarity search solution
108
108
  ]
109
109
  logging = [
110
110
  "wandb", # https://docs.ultralytics.com/integrations/weights-biases/
@@ -115,7 +115,15 @@ extra = [
115
115
  "hub-sdk>=0.0.12", # Ultralytics HUB
116
116
  "ipython", # interactive notebook
117
117
  "albumentations>=1.4.6", # training augmentations
118
- "pycocotools>=2.0.7", # COCO mAP
118
+ "faster-coco-eval>=1.6.7", # COCO mAP
119
+ ]
120
+ typing = [
121
+ "scipy-stubs",
122
+ "types-pillow",
123
+ "types-psutil",
124
+ "types-pyyaml",
125
+ "types-requests",
126
+ "types-shapely",
119
127
  ]
120
128
 
121
129
  [project.urls]
@@ -181,5 +189,5 @@ close-quotes-on-newline = true
181
189
  in-place = true
182
190
 
183
191
  [tool.codespell]
184
- ignore-words-list = "crate,nd,ned,strack,dota,ane,segway,fo,gool,winn,commend,bloc,nam,afterall"
185
- skip = '*.pt,*.pth,*.torchscript,*.onnx,*.tflite,*.pb,*.bin,*.param,*.mlmodel,*.engine,*.npy,*.data*,*.csv,*pnnx*,*venv*,*translat*,__pycache__*,*.ico,*.jpg,*.png,*.mp4,*.mov,/runs,/.git,./docs/??/*.md,./docs/mkdocs_??.yml'
192
+ ignore-words-list = "grey,writeable,finalY,RepResNet,Idenfy,WIT,Smoot,EHR,ROUGE,ALS,iTerm,Carmel,FPR,Hach,Calle,ore,COO,MOT,crate,nd,ned,strack,dota,ane,segway,fo,gool,winn,commend,bloc,nam,afterall,skelton,goin"
193
+ skip = "*.pt,*.pth,*.torchscript,*.onnx,*.tflite,*.pb,*.bin,*.param,*.mlmodel,*.engine,*.npy,*.data*,*.csv,*pnnx*,*venv*,*translat*,*lock*,__pycache__*,*.ico,*.jpg,*.png,*.webp,*.avif,*.mp4,*.mov,/runs,/.git,./docs/??/*.md,./docs/mkdocs_??.yml"
@@ -1,25 +1,23 @@
1
1
  # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
2
2
 
3
3
  from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS
4
- from ultralytics.utils import ASSETS, ROOT, WEIGHTS_DIR, checks
4
+ from ultralytics.utils import ASSETS, WEIGHTS_DIR, checks
5
5
 
6
6
  # Constants used in tests
7
7
  MODEL = WEIGHTS_DIR / "path with spaces" / "yolo11n.pt" # test spaces in path
8
8
  CFG = "yolo11n.yaml"
9
9
  SOURCE = ASSETS / "bus.jpg"
10
10
  SOURCES_LIST = [ASSETS / "bus.jpg", ASSETS, ASSETS / "*", ASSETS / "**/*.jpg"]
11
- TMP = (ROOT / "../tests/tmp").resolve() # temp directory for test files
12
11
  CUDA_IS_AVAILABLE = checks.cuda_is_available()
13
12
  CUDA_DEVICE_COUNT = checks.cuda_device_count()
14
13
  TASK_MODEL_DATA = [(task, WEIGHTS_DIR / TASK2MODEL[task], TASK2DATA[task]) for task in TASKS]
15
- MODELS = frozenset(list(TASK2MODEL.values()) + ["yolo11n-grayscale.pt"])
14
+ MODELS = frozenset([*list(TASK2MODEL.values()), "yolo11n-grayscale.pt"])
16
15
 
17
16
  __all__ = (
18
- "MODEL",
19
17
  "CFG",
18
+ "CUDA_DEVICE_COUNT",
19
+ "CUDA_IS_AVAILABLE",
20
+ "MODEL",
20
21
  "SOURCE",
21
22
  "SOURCES_LIST",
22
- "TMP",
23
- "CUDA_IS_AVAILABLE",
24
- "CUDA_DEVICE_COUNT",
25
23
  )
@@ -3,8 +3,6 @@
3
3
  import shutil
4
4
  from pathlib import Path
5
5
 
6
- from tests import TMP
7
-
8
6
 
9
7
  def pytest_addoption(parser):
10
8
  """Add custom command-line options to pytest."""
@@ -12,8 +10,7 @@ def pytest_addoption(parser):
12
10
 
13
11
 
14
12
  def pytest_collection_modifyitems(config, items):
15
- """
16
- Modify the list of test items to exclude tests marked as slow if the --slow option is not specified.
13
+ """Modify the list of test items to exclude tests marked as slow if the --slow option is not specified.
17
14
 
18
15
  Args:
19
16
  config: The pytest configuration object that provides access to command-line options.
@@ -25,11 +22,10 @@ def pytest_collection_modifyitems(config, items):
25
22
 
26
23
 
27
24
  def pytest_sessionstart(session):
28
- """
29
- Initialize session configurations for pytest.
25
+ """Initialize session configurations for pytest.
30
26
 
31
27
  This function is automatically called by pytest after the 'Session' object has been created but before performing
32
- test collection. It sets the initial seeds and prepares the temporary directory for the test session.
28
+ test collection. It sets the initial seeds for the test session.
33
29
 
34
30
  Args:
35
31
  session: The pytest session object.
@@ -37,16 +33,13 @@ def pytest_sessionstart(session):
37
33
  from ultralytics.utils.torch_utils import init_seeds
38
34
 
39
35
  init_seeds()
40
- shutil.rmtree(TMP, ignore_errors=True) # Delete any existing tests/tmp directory
41
- TMP.mkdir(parents=True, exist_ok=True) # Create a new empty directory
42
36
 
43
37
 
44
38
  def pytest_terminal_summary(terminalreporter, exitstatus, config):
45
- """
46
- Cleanup operations after pytest session.
39
+ """Cleanup operations after pytest session.
47
40
 
48
- This function is automatically called by pytest at the end of the entire test session. It removes certain files
49
- and directories used during testing.
41
+ This function is automatically called by pytest at the end of the entire test session. It removes certain files and
42
+ directories used during testing.
50
43
 
51
44
  Args:
52
45
  terminalreporter: The terminal reporter object used for terminal output.
@@ -56,11 +49,11 @@ def pytest_terminal_summary(terminalreporter, exitstatus, config):
56
49
  from ultralytics.utils import WEIGHTS_DIR
57
50
 
58
51
  # Remove files
59
- models = [path for x in ["*.onnx", "*.torchscript"] for path in WEIGHTS_DIR.rglob(x)]
60
- for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript"] + models:
52
+ models = [path for x in {"*.onnx", "*.torchscript"} for path in WEIGHTS_DIR.rglob(x)]
53
+ for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript", *models]:
61
54
  Path(file).unlink(missing_ok=True)
62
55
 
63
56
  # Remove directories
64
- models = [path for x in ["*.mlpackage", "*_openvino_model"] for path in WEIGHTS_DIR.rglob(x)]
65
- for directory in [WEIGHTS_DIR / "path with spaces", TMP.parents[1] / ".pytest_cache", TMP] + models:
57
+ models = [path for x in {"*.mlpackage", "*_openvino_model"} for path in WEIGHTS_DIR.rglob(x)]
58
+ for directory in [WEIGHTS_DIR / "path with spaces", *models]:
66
59
  shutil.rmtree(directory, ignore_errors=True)
@@ -1,13 +1,14 @@
1
1
  # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
2
2
 
3
3
  import subprocess
4
+ from pathlib import Path
4
5
 
5
6
  import pytest
6
7
  from PIL import Image
7
8
 
8
9
  from tests import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE, MODELS, TASK_MODEL_DATA
9
10
  from ultralytics.utils import ARM64, ASSETS, LINUX, WEIGHTS_DIR, checks
10
- from ultralytics.utils.torch_utils import TORCH_1_9
11
+ from ultralytics.utils.torch_utils import TORCH_1_11
11
12
 
12
13
 
13
14
  def run(cmd: str) -> None:
@@ -33,13 +34,13 @@ def test_train(task: str, model: str, data: str) -> None:
33
34
  @pytest.mark.parametrize("task,model,data", TASK_MODEL_DATA)
34
35
  def test_val(task: str, model: str, data: str) -> None:
35
36
  """Test YOLO validation process for specified task, model, and data using a shell command."""
36
- run(f"yolo val {task} model={model} data={data} imgsz=32 save_txt save_json")
37
+ run(f"yolo val {task} model={model} data={data} imgsz=32 save_txt save_json visualize")
37
38
 
38
39
 
39
40
  @pytest.mark.parametrize("task,model,data", TASK_MODEL_DATA)
40
41
  def test_predict(task: str, model: str, data: str) -> None:
41
42
  """Test YOLO prediction on provided sample assets for specified task and model."""
42
- run(f"yolo {task} predict model={model} source={ASSETS} imgsz=32 save save_crop save_txt")
43
+ run(f"yolo {task} predict model={model} source={ASSETS} imgsz=32 save save_crop save_txt visualize")
43
44
 
44
45
 
45
46
  @pytest.mark.parametrize("model", MODELS)
@@ -48,15 +49,12 @@ def test_export(model: str) -> None:
48
49
  run(f"yolo export model={model} format=torchscript imgsz=32")
49
50
 
50
51
 
51
- def test_rtdetr(task: str = "detect", model: str = "yolov8n-rtdetr.yaml", data: str = "coco8.yaml") -> None:
52
+ @pytest.mark.skipif(not TORCH_1_11, reason="RTDETR requires torch>=1.11")
53
+ def test_rtdetr(task: str = "detect", model: Path = WEIGHTS_DIR / "rtdetr-l.pt", data: str = "coco8.yaml") -> None:
52
54
  """Test the RTDETR functionality within Ultralytics for detection tasks using specified model and data."""
53
- # Warning: must use imgsz=640 (note also add comma, spaces, fraction=0.25 args to test single-image training)
54
- run(f"yolo train {task} model={model} data={data} --imgsz= 160 epochs =1, cache = disk fraction=0.25") # spaces
55
+ # Add comma, spaces, fraction=0.25 args to test single-image training
55
56
  run(f"yolo predict {task} model={model} source={ASSETS / 'bus.jpg'} imgsz=160 save save_crop save_txt")
56
- if TORCH_1_9:
57
- weights = WEIGHTS_DIR / "rtdetr-l.pt"
58
- run(f"yolo predict {task} model={weights} source={ASSETS / 'bus.jpg'} imgsz=160 save save_crop save_txt")
59
- run(f"yolo train {task} model={weights} epochs=1 imgsz=160 cache=disk data=coco8.yaml")
57
+ run(f"yolo train {task} model={model} data={data} --imgsz= 160 epochs =1, cache = disk fraction=0.25")
60
58
 
61
59
 
62
60
  @pytest.mark.skipif(checks.IS_PYTHON_3_12, reason="MobileSAM with CLIP is not supported in Python 3.12")
@@ -84,7 +82,7 @@ def test_fastsam(
84
82
  everything_results = sam_model(s, device="cpu", retina_masks=True, imgsz=320, conf=0.4, iou=0.9)
85
83
 
86
84
  # Remove small regions
87
- new_masks, _ = Predictor.remove_small_regions(everything_results[0].masks.data, min_area=20)
85
+ _new_masks, _ = Predictor.remove_small_regions(everything_results[0].masks.data, min_area=20)
88
86
 
89
87
  # Run inference with bboxes and points and texts prompt at the same time
90
88
  sam_model(source, bboxes=[439, 437, 524, 709], points=[[200, 200]], labels=[1], texts="a photo of a dog")