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.
- {ultralytics-8.3.148 → ultralytics-8.3.226}/PKG-INFO +43 -36
- {ultralytics-8.3.148 → ultralytics-8.3.226}/README.md +26 -26
- {ultralytics-8.3.148 → ultralytics-8.3.226}/pyproject.toml +20 -12
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/__init__.py +5 -7
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/conftest.py +10 -17
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_cli.py +9 -11
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_cuda.py +11 -7
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_engine.py +29 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_exports.py +82 -21
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_integrations.py +11 -14
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_python.py +126 -87
- {ultralytics-8.3.148 → ultralytics-8.3.226}/tests/test_solutions.py +165 -154
- ultralytics-8.3.226/ultralytics/__init__.py +48 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/__init__.py +83 -82
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/Argoverse.yaml +4 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/DOTAv1.5.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/DOTAv1.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/GlobalWheat2020.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/HomeObjects-3K.yaml +4 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/ImageNet.yaml +3 -3
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/Objects365.yaml +24 -20
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/SKU-110K.yaml +9 -9
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/VOC.yaml +10 -13
- ultralytics-8.3.226/ultralytics/cfg/datasets/VisDrone.yaml +87 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/african-wildlife.yaml +5 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/brain-tumor.yaml +4 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/carparts-seg.yaml +5 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco-pose.yaml +26 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco.yaml +4 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco128-seg.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco128.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-grayscale.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-multispectral.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-pose.yaml +23 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8-seg.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/coco8.yaml +2 -2
- ultralytics-8.3.226/ultralytics/cfg/datasets/construction-ppe.yaml +32 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/crack-seg.yaml +5 -5
- ultralytics-8.3.226/ultralytics/cfg/datasets/dog-pose.yaml +52 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/dota8-multispectral.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/dota8.yaml +2 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/hand-keypoints.yaml +29 -4
- ultralytics-8.3.226/ultralytics/cfg/datasets/kitti.yaml +27 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/lvis.yaml +9 -9
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/medical-pills.yaml +4 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/open-images-v7.yaml +3 -3
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/package-seg.yaml +5 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/signature.yaml +4 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/tiger-pose.yaml +20 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/datasets/xView.yaml +5 -5
- ultralytics-8.3.226/ultralytics/cfg/default.yaml +130 -0
- ultralytics-8.3.226/ultralytics/cfg/trackers/botsort.yaml +21 -0
- ultralytics-8.3.226/ultralytics/cfg/trackers/bytetrack.yaml +12 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/annotator.py +10 -11
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/augment.py +499 -517
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/base.py +41 -49
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/build.py +180 -27
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/converter.py +157 -60
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/dataset.py +106 -107
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/loaders.py +60 -70
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/split.py +10 -10
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/split_dota.py +33 -38
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/utils.py +85 -84
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/exporter.py +364 -491
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/model.py +182 -213
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/predictor.py +72 -70
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/results.py +269 -341
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/trainer.py +216 -120
- ultralytics-8.3.226/ultralytics/engine/tuner.py +448 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/validator.py +44 -35
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/__init__.py +22 -22
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/auth.py +8 -14
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/google/__init__.py +17 -19
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/session.py +33 -43
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/hub/utils.py +10 -115
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/model.py +20 -20
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/predict.py +28 -38
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/utils.py +1 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/fastsam/val.py +4 -10
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/model.py +13 -13
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/predict.py +7 -9
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/nas/val.py +1 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/model.py +7 -8
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/predict.py +16 -18
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/train.py +14 -16
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/rtdetr/val.py +47 -54
- ultralytics-8.3.226/ultralytics/models/sam/__init__.py +12 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/amg.py +33 -36
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/build.py +15 -17
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/model.py +26 -32
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/blocks.py +99 -118
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/decoders.py +33 -40
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/encoders.py +97 -113
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/memory_attention.py +31 -36
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/sam.py +69 -79
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/tiny_encoder.py +103 -121
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/transformer.py +46 -53
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/utils.py +47 -56
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/predict.py +678 -291
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/loss.py +65 -75
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/ops.py +28 -30
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/predict.py +16 -25
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/train.py +32 -67
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/val.py +63 -51
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/predict.py +11 -15
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/train.py +55 -46
- ultralytics-8.3.226/ultralytics/models/yolo/detect/val.py +507 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/model.py +88 -96
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/predict.py +5 -8
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/train.py +17 -27
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/val.py +94 -99
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/predict.py +6 -9
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/pose/train.py +16 -62
- ultralytics-8.3.226/ultralytics/models/yolo/pose/val.py +259 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/segment/predict.py +17 -21
- ultralytics-8.3.226/ultralytics/models/yolo/segment/train.py +69 -0
- ultralytics-8.3.226/ultralytics/models/yolo/segment/val.py +249 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/train.py +35 -37
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/train_world.py +40 -29
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/predict.py +22 -29
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/train.py +47 -81
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/train_seg.py +5 -10
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/val.py +29 -38
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/__init__.py +9 -11
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/autobackend.py +134 -106
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/__init__.py +63 -63
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/activation.py +4 -6
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/block.py +145 -233
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/conv.py +58 -102
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/head.py +109 -155
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/transformer.py +76 -105
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/modules/utils.py +15 -20
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/tasks.py +132 -224
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/nn/text_model.py +68 -49
- ultralytics-8.3.226/ultralytics/py.typed +1 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/ai_gym.py +17 -23
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/analytics.py +19 -18
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/config.py +22 -21
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/distance_calculation.py +16 -19
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/heatmap.py +16 -18
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/instance_segmentation.py +14 -15
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_blurrer.py +13 -14
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_counter.py +35 -40
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/object_cropper.py +10 -11
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/parking_management.py +35 -33
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/queue_management.py +13 -13
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/region_counter.py +53 -47
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/security_alarm.py +22 -25
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/similarity_search.py +39 -50
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/solutions.py +211 -261
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/speed_estimation.py +9 -11
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/streamlit_inference.py +91 -32
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/templates/similarity-search.html +31 -24
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/trackzone.py +20 -20
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/vision_eye.py +11 -12
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/__init__.py +1 -1
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/basetrack.py +2 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/bot_sort.py +27 -30
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/byte_tracker.py +54 -60
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/track.py +3 -8
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/gmc.py +13 -18
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/kalman_filter.py +35 -43
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/matching.py +12 -16
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/__init__.py +158 -350
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/autobatch.py +5 -6
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/autodevice.py +23 -25
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/benchmarks.py +119 -97
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/base.py +11 -12
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/clearml.py +5 -6
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/comet.py +90 -59
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/dvc.py +13 -18
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/hub.py +3 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/mlflow.py +4 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/neptune.py +4 -6
- ultralytics-8.3.226/ultralytics/utils/callbacks/platform.py +73 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/raytune.py +3 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/tensorboard.py +6 -8
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/wb.py +19 -16
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/checks.py +130 -98
- ultralytics-8.3.226/ultralytics/utils/cpu.py +85 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/dist.py +20 -16
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/downloads.py +98 -98
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/errors.py +6 -8
- ultralytics-8.3.226/ultralytics/utils/events.py +113 -0
- ultralytics-8.3.226/ultralytics/utils/export/__init__.py +7 -0
- ultralytics-8.3.148/ultralytics/utils/export.py → ultralytics-8.3.226/ultralytics/utils/export/engine.py +40 -36
- ultralytics-8.3.226/ultralytics/utils/export/imx.py +294 -0
- ultralytics-8.3.226/ultralytics/utils/export/tensorflow.py +217 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/files.py +19 -23
- ultralytics-8.3.226/ultralytics/utils/git.py +136 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/instance.py +68 -79
- ultralytics-8.3.226/ultralytics/utils/logger.py +404 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/loss.py +35 -34
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/metrics.py +530 -519
- ultralytics-8.3.226/ultralytics/utils/nms.py +337 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/ops.py +164 -348
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/patches.py +62 -21
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/plotting.py +183 -198
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/tal.py +28 -44
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/torch_utils.py +195 -208
- ultralytics-8.3.226/ultralytics/utils/tqdm.py +436 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/triton.py +20 -21
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/tuner.py +6 -5
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/PKG-INFO +43 -36
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/SOURCES.txt +15 -2
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/requires.txt +21 -9
- ultralytics-8.3.148/ultralytics/__init__.py +0 -30
- ultralytics-8.3.148/ultralytics/cfg/datasets/VisDrone.yaml +0 -77
- ultralytics-8.3.148/ultralytics/cfg/datasets/dog-pose.yaml +0 -24
- ultralytics-8.3.148/ultralytics/cfg/default.yaml +0 -127
- ultralytics-8.3.148/ultralytics/cfg/trackers/botsort.yaml +0 -22
- ultralytics-8.3.148/ultralytics/cfg/trackers/bytetrack.yaml +0 -14
- ultralytics-8.3.148/ultralytics/engine/tuner.py +0 -246
- ultralytics-8.3.148/ultralytics/models/sam/__init__.py +0 -6
- ultralytics-8.3.148/ultralytics/models/yolo/detect/val.py +0 -443
- ultralytics-8.3.148/ultralytics/models/yolo/pose/val.py +0 -417
- ultralytics-8.3.148/ultralytics/models/yolo/segment/train.py +0 -127
- ultralytics-8.3.148/ultralytics/models/yolo/segment/val.py +0 -436
- {ultralytics-8.3.148 → ultralytics-8.3.226}/LICENSE +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/setup.cfg +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/assets/bus.jpg +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/assets/zidane.jpg +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-cls-resnet18.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-cls.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-obb.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-pose.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yolo11.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yoloe-11-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/11/yoloe-11.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-cls.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-obb.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-pose.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/12/yolo12.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/rt-detr/rtdetr-x.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10b.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10l.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10m.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10n.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10s.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v10/yolov10x.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3-spp.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3-tiny.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v3/yolov3.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v5/yolov5-p6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v5/yolov5.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v6/yolov6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yoloe-v8-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yoloe-v8.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls-resnet101.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls-resnet50.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-cls.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost-p2.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost-p6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-ghost.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-obb.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-p2.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-p6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-pose-p6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-pose.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-rtdetr.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-seg-p6.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-world.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8-worldv2.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v8/yolov8.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9c-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9c.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9e-seg.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9e.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9m.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9s.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/cfg/models/v9/yolov9t.yaml +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/__init__.py +4 -4
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/download_weights.sh +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_coco.sh +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_coco128.sh +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/data/scripts/get_imagenet.sh +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/engine/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/sam/modules/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/utils/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/classify/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/detect/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/obb/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/segment/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/world/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/models/yolo/yoloe/__init__.py +7 -7
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/solutions/__init__.py +12 -12
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/trackers/utils/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics/utils/callbacks/__init__.py +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/dependency_links.txt +0 -0
- {ultralytics-8.3.148 → ultralytics-8.3.226}/ultralytics.egg-info/entry_points.txt +0 -0
- {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.
|
|
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:
|
|
48
|
-
Requires-Dist:
|
|
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.
|
|
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:
|
|
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
|
|
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:
|
|
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/
|
|
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://
|
|
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
|
-
[](https://pypi.org/project/ultralytics/) [](https://
|
|
150
|
+
[](https://pypi.org/project/ultralytics/) [](https://clickpy.clickhouse.com/dashboard/ultralytics) [](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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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 |
|
|
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 |
|
|
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.
|
|
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 |
|
|
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.
|
|
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 |
|
|
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/
|
|
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://
|
|
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
|
-
[](https://pypi.org/project/ultralytics/) [](https://
|
|
60
|
+
[](https://pypi.org/project/ultralytics/) [](https://clickpy.clickhouse.com/dashboard/ultralytics) [](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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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 |
|
|
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.
|
|
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.
|
|
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.
|
|
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.
|
|
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 |
|
|
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 |
|
|
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.
|
|
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 |
|
|
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.
|
|
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 |
|
|
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
|
-
"
|
|
77
|
-
"
|
|
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.
|
|
92
|
-
"mkdocs-macros-plugin>=1.0.5"
|
|
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
|
-
"
|
|
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
|
-
"
|
|
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 =
|
|
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,
|
|
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())
|
|
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
|
|
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
|
-
|
|
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
|
|
60
|
-
for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript"
|
|
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
|
|
65
|
-
for directory in [WEIGHTS_DIR / "path with spaces",
|
|
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
|
|
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
|
-
|
|
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
|
-
#
|
|
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
|
-
|
|
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
|
-
|
|
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")
|