ultralytics 8.3.123__tar.gz → 8.3.124__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.123/ultralytics.egg-info → ultralytics-8.3.124}/PKG-INFO +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/__init__.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/__init__.py +3 -9
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/base.py +1 -2
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/loaders.py +3 -4
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/utils.py +1 -2
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/model.py +4 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/model.py +2 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/tiny_encoder.py +2 -3
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/predict.py +4 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/tasks.py +5 -5
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/analytics.py +2 -2
- ultralytics-8.3.124/ultralytics/solutions/config.py +104 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/heatmap.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/object_blurrer.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/object_cropper.py +2 -2
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/parking_management.py +2 -2
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/security_alarm.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/solutions.py +6 -9
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/speed_estimation.py +4 -4
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/trackzone.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/vision_eye.py +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/__init__.py +1 -3
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/checks.py +2 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/dist.py +2 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/downloads.py +6 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/torch_utils.py +10 -5
- {ultralytics-8.3.123 → ultralytics-8.3.124/ultralytics.egg-info}/PKG-INFO +1 -1
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics.egg-info/SOURCES.txt +1 -1
- ultralytics-8.3.123/ultralytics/cfg/solutions/default.yaml +0 -24
- {ultralytics-8.3.123 → ultralytics-8.3.124}/LICENSE +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/README.md +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/pyproject.toml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/setup.cfg +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/conftest.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_cli.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_cuda.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_engine.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_exports.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_integrations.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_python.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/tests/test_solutions.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/assets/bus.jpg +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/assets/zidane.jpg +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/Argoverse.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/DOTAv1.5.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/DOTAv1.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/GlobalWheat2020.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/ImageNet.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/Objects365.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/SKU-110K.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/VOC.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/VisDrone.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/african-wildlife.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/brain-tumor.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/carparts-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco128-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco128.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco8-multispectral.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco8-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco8-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/coco8.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/crack-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/dog-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/dota8-multispectral.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/dota8.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/hand-keypoints.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/lvis.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/medical-pills.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/open-images-v7.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/package-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/signature.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/tiger-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/datasets/xView.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/default.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11-cls-resnet18.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11-cls.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11-obb.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yolo11.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yoloe-11-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/11/yoloe-11.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/12/yolo12-cls.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/12/yolo12-obb.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/12/yolo12-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/12/yolo12-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/12/yolo12.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/rt-detr/rtdetr-l.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/rt-detr/rtdetr-x.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10b.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10l.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10m.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10n.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10s.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v10/yolov10x.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v3/yolov3-spp.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v3/yolov3-tiny.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v3/yolov3.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v5/yolov5-p6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v5/yolov5.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v6/yolov6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yoloe-v8-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yoloe-v8.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-cls-resnet101.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-cls-resnet50.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-cls.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-ghost-p2.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-ghost-p6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-ghost.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-obb.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-p2.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-p6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-pose-p6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-pose.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-rtdetr.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-seg-p6.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-world.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8-worldv2.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v8/yolov8.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9c-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9c.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9e-seg.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9e.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9m.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9s.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/models/v9/yolov9t.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/trackers/botsort.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/cfg/trackers/bytetrack.yaml +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/annotator.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/augment.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/build.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/converter.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/dataset.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/scripts/download_weights.sh +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/scripts/get_coco.sh +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/scripts/get_coco128.sh +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/scripts/get_imagenet.sh +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/split.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/data/split_dota.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/exporter.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/predictor.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/results.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/trainer.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/tuner.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/engine/validator.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/hub/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/hub/auth.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/hub/google/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/hub/session.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/hub/utils.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/fastsam/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/fastsam/model.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/fastsam/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/fastsam/utils.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/fastsam/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/nas/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/nas/model.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/nas/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/nas/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/rtdetr/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/rtdetr/model.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/rtdetr/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/rtdetr/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/rtdetr/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/amg.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/build.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/blocks.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/decoders.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/encoders.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/memory_attention.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/sam.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/transformer.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/sam/modules/utils.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/utils/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/utils/loss.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/utils/ops.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/classify/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/classify/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/classify/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/classify/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/detect/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/detect/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/detect/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/detect/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/model.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/obb/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/obb/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/obb/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/obb/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/pose/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/pose/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/pose/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/pose/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/segment/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/segment/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/segment/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/segment/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/world/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/world/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/world/train_world.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/yoloe/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/yoloe/predict.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/yoloe/train.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/yoloe/train_seg.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/models/yolo/yoloe/val.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/autobackend.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/activation.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/block.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/conv.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/head.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/transformer.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/modules/utils.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/nn/text_model.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/ai_gym.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/distance_calculation.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/instance_segmentation.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/object_counter.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/queue_management.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/region_counter.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/solutions/streamlit_inference.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/basetrack.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/bot_sort.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/byte_tracker.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/track.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/utils/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/utils/gmc.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/utils/kalman_filter.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/trackers/utils/matching.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/autobatch.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/benchmarks.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/__init__.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/base.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/clearml.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/comet.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/dvc.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/hub.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/mlflow.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/neptune.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/raytune.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/tensorboard.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/callbacks/wb.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/errors.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/export.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/files.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/instance.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/loss.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/metrics.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/ops.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/patches.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/plotting.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/tal.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/triton.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics/utils/tuner.py +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics.egg-info/dependency_links.txt +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics.egg-info/entry_points.txt +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/ultralytics.egg-info/requires.txt +0 -0
- {ultralytics-8.3.123 → ultralytics-8.3.124}/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.124
|
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>
|
@@ -15,7 +15,6 @@ from ultralytics.utils import (
|
|
15
15
|
DEFAULT_CFG,
|
16
16
|
DEFAULT_CFG_DICT,
|
17
17
|
DEFAULT_CFG_PATH,
|
18
|
-
DEFAULT_SOL_DICT,
|
19
18
|
IS_VSCODE,
|
20
19
|
LOGGER,
|
21
20
|
RANK,
|
@@ -650,7 +649,6 @@ def handle_yolo_solutions(args: List[str]) -> None:
|
|
650
649
|
>>> handle_yolo_solutions(["inference", "model=yolo11n.pt"])
|
651
650
|
|
652
651
|
Notes:
|
653
|
-
- Default configurations are merged from DEFAULT_SOL_DICT and DEFAULT_CFG_DICT
|
654
652
|
- Arguments can be provided in the format 'key=value' or as boolean flags
|
655
653
|
- Available solutions are defined in SOLUTION_MAP with their respective classes and methods
|
656
654
|
- If an invalid solution is provided, defaults to 'count' solution
|
@@ -662,13 +660,9 @@ def handle_yolo_solutions(args: List[str]) -> None:
|
|
662
660
|
- The inference solution will be launched using the 'streamlit run' command.
|
663
661
|
- The Streamlit app file is located in the Ultralytics package directory.
|
664
662
|
"""
|
665
|
-
|
666
|
-
|
667
|
-
|
668
|
-
"blur_ratio": 0.5,
|
669
|
-
"vision_point": (20, 20),
|
670
|
-
"crop_dir": "cropped-detections",
|
671
|
-
} # arguments dictionary
|
663
|
+
from ultralytics.solutions.config import SolutionConfig
|
664
|
+
|
665
|
+
full_args_dict = vars(SolutionConfig()) # arguments dictionary
|
672
666
|
overrides = {}
|
673
667
|
|
674
668
|
# check dictionary alignment
|
@@ -11,7 +11,6 @@ from typing import Optional
|
|
11
11
|
|
12
12
|
import cv2
|
13
13
|
import numpy as np
|
14
|
-
import psutil
|
15
14
|
from torch.utils.data import Dataset
|
16
15
|
|
17
16
|
from ultralytics.data.utils import FORMATS_HELP_MSG, HELP_URL, IMG_FORMATS, check_file_speeds
|
@@ -336,7 +335,7 @@ class BaseDataset(Dataset):
|
|
336
335
|
ratio = self.imgsz / max(im.shape[0], im.shape[1]) # max(h, w) # ratio
|
337
336
|
b += im.nbytes * ratio**2
|
338
337
|
mem_required = b * self.ni / n * (1 + safety_margin) # GB required to cache dataset into RAM
|
339
|
-
mem = psutil.virtual_memory()
|
338
|
+
mem = __import__("psutil").virtual_memory()
|
340
339
|
if mem_required > mem.available:
|
341
340
|
self.cache = None
|
342
341
|
LOGGER.warning(
|
@@ -4,14 +4,13 @@ import glob
|
|
4
4
|
import math
|
5
5
|
import os
|
6
6
|
import time
|
7
|
+
import urllib
|
7
8
|
from dataclasses import dataclass
|
8
9
|
from pathlib import Path
|
9
10
|
from threading import Thread
|
10
|
-
from urllib.parse import urlparse
|
11
11
|
|
12
12
|
import cv2
|
13
13
|
import numpy as np
|
14
|
-
import requests
|
15
14
|
import torch
|
16
15
|
from PIL import Image
|
17
16
|
|
@@ -111,7 +110,7 @@ class LoadStreams:
|
|
111
110
|
for i, s in enumerate(sources): # index, source
|
112
111
|
# Start thread to read frames from video stream
|
113
112
|
st = f"{i + 1}/{n}: {s}... "
|
114
|
-
if urlparse(s).hostname in {"www.youtube.com", "youtube.com", "youtu.be"}: #
|
113
|
+
if urllib.parse.urlparse(s).hostname in {"www.youtube.com", "youtube.com", "youtu.be"}: # YouTube video
|
115
114
|
# YouTube format i.e. 'https://www.youtube.com/watch?v=Jsn8D3aC840' or 'https://youtu.be/Jsn8D3aC840'
|
116
115
|
s = get_best_youtube_url(s)
|
117
116
|
s = eval(s) if s.isnumeric() else s # i.e. s = '0' local webcam
|
@@ -589,7 +588,7 @@ def autocast_list(source):
|
|
589
588
|
files = []
|
590
589
|
for im in source:
|
591
590
|
if isinstance(im, (str, Path)): # filename or uri
|
592
|
-
files.append(Image.open(
|
591
|
+
files.append(Image.open(urllib.request.urlopen(im) if str(im).startswith("http") else im))
|
593
592
|
elif isinstance(im, (Image.Image, np.ndarray)): # PIL or np Image
|
594
593
|
files.append(im)
|
595
594
|
else:
|
@@ -1,6 +1,5 @@
|
|
1
1
|
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
2
2
|
|
3
|
-
import hashlib
|
4
3
|
import json
|
5
4
|
import os
|
6
5
|
import random
|
@@ -131,7 +130,7 @@ def get_hash(paths):
|
|
131
130
|
size += os.stat(p).st_size
|
132
131
|
except OSError:
|
133
132
|
continue
|
134
|
-
h = hashlib.sha256(str(size).encode()) # hash sizes
|
133
|
+
h = __import__("hashlib").sha256(str(size).encode()) # hash sizes
|
135
134
|
h.update("".join(paths).encode()) # hash paths
|
136
135
|
return h.hexdigest() # return hash
|
137
136
|
|
@@ -10,7 +10,6 @@ from PIL import Image
|
|
10
10
|
|
11
11
|
from ultralytics.cfg import TASK2DATA, get_cfg, get_save_dir
|
12
12
|
from ultralytics.engine.results import Results
|
13
|
-
from ultralytics.hub import HUB_WEB_ROOT, HUBTrainingSession
|
14
13
|
from ultralytics.nn.tasks import attempt_load_one_weight, guess_model_task, yaml_model_load
|
15
14
|
from ultralytics.utils import (
|
16
15
|
ARGV,
|
@@ -126,6 +125,8 @@ class Model(torch.nn.Module):
|
|
126
125
|
|
127
126
|
# Check if Ultralytics HUB model from https://hub.ultralytics.com
|
128
127
|
if self.is_hub_model(model):
|
128
|
+
from ultralytics.hub import HUBTrainingSession
|
129
|
+
|
129
130
|
# Fetch model from HUB
|
130
131
|
checks.check_requirements("hub-sdk>=0.0.12")
|
131
132
|
session = HUBTrainingSession.create_session(model)
|
@@ -225,6 +226,8 @@ class Model(torch.nn.Module):
|
|
225
226
|
>>> Model.is_hub_model("yolo11n.pt")
|
226
227
|
False
|
227
228
|
"""
|
229
|
+
from ultralytics.hub import HUB_WEB_ROOT
|
230
|
+
|
228
231
|
return model.startswith(f"{HUB_WEB_ROOT}/models/")
|
229
232
|
|
230
233
|
def _new(self, cfg: str, task=None, model=None, verbose=False) -> None:
|
@@ -19,7 +19,6 @@ from pathlib import Path
|
|
19
19
|
from ultralytics.engine.model import Model
|
20
20
|
from ultralytics.utils.torch_utils import model_info
|
21
21
|
|
22
|
-
from .build import build_sam
|
23
22
|
from .predict import Predictor, SAM2Predictor
|
24
23
|
|
25
24
|
|
@@ -78,6 +77,8 @@ class SAM(Model):
|
|
78
77
|
>>> sam = SAM("sam_b.pt")
|
79
78
|
>>> sam._load("path/to/custom_weights.pt")
|
80
79
|
"""
|
80
|
+
from .build import build_sam # slow import
|
81
|
+
|
81
82
|
self.model = build_sam(weights)
|
82
83
|
|
83
84
|
def predict(self, source, stream=False, bboxes=None, points=None, labels=None, **kwargs):
|
@@ -15,7 +15,6 @@ from typing import Tuple
|
|
15
15
|
import torch
|
16
16
|
import torch.nn as nn
|
17
17
|
import torch.nn.functional as F
|
18
|
-
import torch.utils.checkpoint as checkpoint
|
19
18
|
|
20
19
|
from ultralytics.nn.modules import LayerNorm2d
|
21
20
|
from ultralytics.utils.instance import to_2tuple
|
@@ -308,7 +307,7 @@ class ConvLayer(nn.Module):
|
|
308
307
|
def forward(self, x):
|
309
308
|
"""Processes input through convolutional layers, applying MBConv blocks and optional downsampling."""
|
310
309
|
for blk in self.blocks:
|
311
|
-
x =
|
310
|
+
x = torch.utils.checkpoint(blk, x) if self.use_checkpoint else blk(x) # warn: checkpoint is slow import
|
312
311
|
return x if self.downsample is None else self.downsample(x)
|
313
312
|
|
314
313
|
|
@@ -751,7 +750,7 @@ class BasicLayer(nn.Module):
|
|
751
750
|
def forward(self, x):
|
752
751
|
"""Processes input through TinyViT blocks and optional downsampling."""
|
753
752
|
for blk in self.blocks:
|
754
|
-
x =
|
753
|
+
x = torch.utils.checkpoint(blk, x) if self.use_checkpoint else blk(x) # warn: checkpoint is slow import
|
755
754
|
return x if self.downsample is None else self.downsample(x)
|
756
755
|
|
757
756
|
def extra_repr(self) -> str:
|
@@ -31,7 +31,6 @@ from .amg import (
|
|
31
31
|
uncrop_boxes_xyxy,
|
32
32
|
uncrop_masks,
|
33
33
|
)
|
34
|
-
from .build import build_sam
|
35
34
|
|
36
35
|
|
37
36
|
class Predictor(BasePredictor):
|
@@ -439,6 +438,8 @@ class Predictor(BasePredictor):
|
|
439
438
|
|
440
439
|
def get_model(self):
|
441
440
|
"""Retrieves or builds the Segment Anything Model (SAM) for image segmentation tasks."""
|
441
|
+
from .build import build_sam # slow import
|
442
|
+
|
442
443
|
return build_sam(self.args.model)
|
443
444
|
|
444
445
|
def postprocess(self, preds, img, orig_imgs):
|
@@ -658,6 +659,8 @@ class SAM2Predictor(Predictor):
|
|
658
659
|
|
659
660
|
def get_model(self):
|
660
661
|
"""Retrieves and initializes the Segment Anything Model 2 (SAM2) for image segmentation tasks."""
|
662
|
+
from .build import build_sam # slow import
|
663
|
+
|
661
664
|
return build_sam(self.args.model)
|
662
665
|
|
663
666
|
def prompt_inference(
|
@@ -92,11 +92,6 @@ from ultralytics.utils.torch_utils import (
|
|
92
92
|
time_sync,
|
93
93
|
)
|
94
94
|
|
95
|
-
try:
|
96
|
-
import thop
|
97
|
-
except ImportError:
|
98
|
-
thop = None # conda support without 'ultralytics-thop' installed
|
99
|
-
|
100
95
|
|
101
96
|
class BaseModel(torch.nn.Module):
|
102
97
|
"""The BaseModel class serves as a base class for all the models in the Ultralytics YOLO family."""
|
@@ -183,6 +178,11 @@ class BaseModel(torch.nn.Module):
|
|
183
178
|
x (torch.Tensor): The input data to the layer.
|
184
179
|
dt (list): A list to store the computation time of the layer.
|
185
180
|
"""
|
181
|
+
try:
|
182
|
+
import thop
|
183
|
+
except ImportError:
|
184
|
+
thop = None # conda support without 'ultralytics-thop' installed
|
185
|
+
|
186
186
|
c = m == self.model[-1] and isinstance(x, list) # is final layer list, copy input as inplace fix
|
187
187
|
flops = thop.profile(m, inputs=[x.copy() if c else x], verbose=False)[0] / 1e9 * 2 if thop else 0 # GFLOPs
|
188
188
|
t = time_sync()
|
@@ -51,7 +51,7 @@ class Analytics(BaseSolution):
|
|
51
51
|
"""Initialize Analytics class with various chart types for visual data representation."""
|
52
52
|
super().__init__(**kwargs)
|
53
53
|
|
54
|
-
self.type = self.CFG["analytics_type"] #
|
54
|
+
self.type = self.CFG["analytics_type"] # type of analytics i.e "line", "pie", "bar" or "area" charts.
|
55
55
|
self.x_label = "Classes" if self.type in {"bar", "pie"} else "Frame#"
|
56
56
|
self.y_label = "Total Counts"
|
57
57
|
|
@@ -61,7 +61,7 @@ class Analytics(BaseSolution):
|
|
61
61
|
self.title = "Ultralytics Solutions" # window name
|
62
62
|
self.max_points = 45 # maximum points to be drawn on window
|
63
63
|
self.fontsize = 25 # text font size for display
|
64
|
-
figsize = (12.8, 7.2)
|
64
|
+
figsize = self.CFG["figsize"] # set output image size i.e (12.8, 7.2) -> w = 1280, h = 720
|
65
65
|
self.color_cycle = cycle(["#DD00BA", "#042AFF", "#FF4447", "#7D24FF", "#BD00FF"])
|
66
66
|
|
67
67
|
self.total_counts = 0 # count variable for storing total counts i.e. for line
|
@@ -0,0 +1,104 @@
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
2
|
+
|
3
|
+
from dataclasses import dataclass, field
|
4
|
+
from typing import List, Optional, Tuple
|
5
|
+
|
6
|
+
import cv2
|
7
|
+
|
8
|
+
|
9
|
+
@dataclass
|
10
|
+
class SolutionConfig:
|
11
|
+
"""
|
12
|
+
Manages configuration parameters for Ultralytics Vision AI solutions.
|
13
|
+
|
14
|
+
The SolutionConfig class serves as a centralized configuration container for all the
|
15
|
+
Ultralytics solution modules: https://docs.ultralytics.com/solutions/#solutions.
|
16
|
+
It leverages Python `dataclass` for clear, type-safe, and maintainable parameter definitions.
|
17
|
+
|
18
|
+
Attributes:
|
19
|
+
source (Optional[str]): Path to the input source (video, RTSP, etc.). Only usable with Solutions CLI.
|
20
|
+
model (Optional[str]): Path to the Ultralytics YOLO model to be used for inference.
|
21
|
+
classes (Optional[List[int]]): List of class indices to filter detections.
|
22
|
+
show_conf (bool): Whether to show confidence scores on the visual output.
|
23
|
+
show_labels (bool): Whether to display class labels on visual output.
|
24
|
+
region (Optional[List[Tuple[int, int]]]): Polygonal region or line for object counting.
|
25
|
+
colormap (Optional[int]): OpenCV colormap constant for visual overlays (e.g., cv2.COLORMAP_JET).
|
26
|
+
show_in (bool): Whether to display count number for objects entering the region.
|
27
|
+
show_out (bool): Whether to display count number for objects leaving the region.
|
28
|
+
up_angle (float): Upper angle threshold used in pose-based workouts monitoring.
|
29
|
+
down_angle (int): Lower angle threshold used in pose-based workouts monitoring.
|
30
|
+
kpts (List[int]): Keypoint indices to monitor, e.g., for pose analytics.
|
31
|
+
analytics_type (str): Type of analytics to perform ("line", "area", "bar", "pie", etc.).
|
32
|
+
figsize (Optional[Tuple[int, int]]): Size of the matplotlib figure used for analytical plots (width, height).
|
33
|
+
blur_ratio (float): Ratio used to blur objects in the video frames (0.0 to 1.0).
|
34
|
+
vision_point (Tuple[int, int]): Reference point for directional tracking or perspective drawing.
|
35
|
+
crop_dir (str): Directory path to save cropped detection images.
|
36
|
+
json_file (str): Path to a JSON file containing data for parking areas.
|
37
|
+
line_width (int): Width for visual display i.e. bounding boxes, keypoints, counts.
|
38
|
+
records (int): Number of detection records to send email alerts.
|
39
|
+
fps (float): Frame rate (Frames Per Second) for speed estimation calculation.
|
40
|
+
max_hist (int): Maximum number of historical points or states stored per tracked object for speed estimation.
|
41
|
+
meter_per_pixel (float): Scale for real-world measurement, used in speed or distance calculations.
|
42
|
+
max_speed (int): Maximum speed limit (e.g., km/h or mph) used in visual alerts or constraints.
|
43
|
+
show (bool): Whether to display the visual output on screen.
|
44
|
+
iou (float): Intersection-over-Union threshold for detection filtering.
|
45
|
+
conf (float): Confidence threshold for keeping predictions.
|
46
|
+
device (Optional[str]): Device to run inference on (e.g., 'cpu', '0' for CUDA GPU).
|
47
|
+
max_det (int): Maximum number of detections allowed per video frame.
|
48
|
+
half (bool): Whether to use FP16 precision (requires a supported CUDA device).
|
49
|
+
tracker (str): Path to tracking configuration YAML file (e.g., 'botsort.yaml').
|
50
|
+
verbose (bool): Enable verbose logging output for debugging or diagnostics.
|
51
|
+
|
52
|
+
Methods:
|
53
|
+
update: Update the configuration with user-defined keyword arguments and raise error on invalid keys.
|
54
|
+
|
55
|
+
Examples:
|
56
|
+
>>> from ultralytics.solutions.config import SolutionConfig
|
57
|
+
>>> cfg = SolutionConfig(model="yolo11n.pt", region=[(0, 0), (100, 0), (100, 100), (0, 100)])
|
58
|
+
>>> cfg.update(show=False, conf=0.3)
|
59
|
+
>>> print(cfg.model)
|
60
|
+
"""
|
61
|
+
|
62
|
+
source: Optional[str] = None
|
63
|
+
model: Optional[str] = None
|
64
|
+
classes: Optional[List[int]] = None
|
65
|
+
show_conf: bool = True
|
66
|
+
show_labels: bool = True
|
67
|
+
region: Optional[List[Tuple[int, int]]] = None
|
68
|
+
colormap: Optional[int] = cv2.COLORMAP_DEEPGREEN
|
69
|
+
show_in: bool = True
|
70
|
+
show_out: bool = True
|
71
|
+
up_angle: float = 145.0
|
72
|
+
down_angle: int = 90
|
73
|
+
kpts: List[int] = field(default_factory=lambda: [6, 8, 10])
|
74
|
+
analytics_type: str = "line"
|
75
|
+
figsize: Optional[Tuple[int, int]] = (12.8, 7.2)
|
76
|
+
blur_ratio: float = 0.5
|
77
|
+
vision_point: Tuple[int, int] = (20, 20)
|
78
|
+
crop_dir: str = "cropped-detections"
|
79
|
+
json_file: str = None
|
80
|
+
line_width: int = 2
|
81
|
+
records: int = 5
|
82
|
+
fps: float = 30.0
|
83
|
+
max_hist: int = 5
|
84
|
+
meter_per_pixel: float = 0.05
|
85
|
+
max_speed: int = 120
|
86
|
+
show: bool = False
|
87
|
+
iou: float = 0.7
|
88
|
+
conf: float = 0.25
|
89
|
+
device: Optional[str] = None
|
90
|
+
max_det: int = 300
|
91
|
+
half: bool = False
|
92
|
+
tracker: str = "botsort.yaml"
|
93
|
+
verbose: bool = True
|
94
|
+
|
95
|
+
def update(self, **kwargs):
|
96
|
+
"""Update configuration parameters with new values provided as keyword arguments."""
|
97
|
+
for key, value in kwargs.items():
|
98
|
+
if hasattr(self, key):
|
99
|
+
setattr(self, key, value)
|
100
|
+
else:
|
101
|
+
raise ValueError(
|
102
|
+
f"❌ {key} is not a valid solution argument, available arguments here: https://docs.ultralytics.com/solutions/#solutions-arguments"
|
103
|
+
)
|
104
|
+
return self
|
@@ -45,7 +45,7 @@ class Heatmap(ObjectCounter):
|
|
45
45
|
self.initialize_region()
|
46
46
|
|
47
47
|
# Store colormap
|
48
|
-
self.colormap =
|
48
|
+
self.colormap = self.CFG["colormap"]
|
49
49
|
self.heatmap = None
|
50
50
|
|
51
51
|
def heatmap_effect(self, box):
|
@@ -41,7 +41,7 @@ class ObjectBlurrer(BaseSolution):
|
|
41
41
|
blur_ratio (float): Intensity of the blur effect (0.1-1.0, default=0.5).
|
42
42
|
"""
|
43
43
|
super().__init__(**kwargs)
|
44
|
-
blur_ratio =
|
44
|
+
blur_ratio = self.CFG["blur_ratio"]
|
45
45
|
if blur_ratio < 0.1:
|
46
46
|
LOGGER.warning("blur ratio cannot be less than 0.1, updating it to default value 0.5")
|
47
47
|
blur_ratio = 0.5
|
@@ -40,7 +40,7 @@ class ObjectCropper(BaseSolution):
|
|
40
40
|
"""
|
41
41
|
super().__init__(**kwargs)
|
42
42
|
|
43
|
-
self.crop_dir =
|
43
|
+
self.crop_dir = self.CFG["crop_dir"] # Directory for storing cropped detections
|
44
44
|
if not os.path.exists(self.crop_dir):
|
45
45
|
os.mkdir(self.crop_dir) # Create directory if it does not exist
|
46
46
|
if self.CFG["show"]:
|
@@ -49,7 +49,7 @@ class ObjectCropper(BaseSolution):
|
|
49
49
|
)
|
50
50
|
self.crop_idx = 0 # Initialize counter for total cropped objects
|
51
51
|
self.iou = self.CFG["iou"]
|
52
|
-
self.conf = self.CFG["conf"]
|
52
|
+
self.conf = self.CFG["conf"]
|
53
53
|
|
54
54
|
def process(self, im0):
|
55
55
|
"""
|
@@ -201,9 +201,9 @@ class ParkingManagement(BaseSolution):
|
|
201
201
|
"""Initialize the parking management system with a YOLO model and visualization settings."""
|
202
202
|
super().__init__(**kwargs)
|
203
203
|
|
204
|
-
self.json_file = self.CFG["json_file"] # Load JSON data
|
204
|
+
self.json_file = self.CFG["json_file"] # Load parking regions JSON data
|
205
205
|
if self.json_file is None:
|
206
|
-
LOGGER.warning("
|
206
|
+
LOGGER.warning("json_file argument missing. Parking region details required.")
|
207
207
|
raise ValueError("❌ Json file path can not be empty")
|
208
208
|
|
209
209
|
with open(self.json_file) as f:
|
@@ -143,7 +143,7 @@ class SecurityAlarm(BaseSolution):
|
|
143
143
|
annotator.box_label(box, label=self.names[cls], color=colors(cls, True))
|
144
144
|
|
145
145
|
total_det = len(self.clss)
|
146
|
-
if total_det
|
146
|
+
if total_det >= self.records and not self.email_sent: # Only send email if not sent before
|
147
147
|
self.send_email(im0, total_det)
|
148
148
|
self.email_sent = True
|
149
149
|
|
@@ -7,7 +7,8 @@ import cv2
|
|
7
7
|
import numpy as np
|
8
8
|
|
9
9
|
from ultralytics import YOLO
|
10
|
-
from ultralytics.
|
10
|
+
from ultralytics.solutions.config import SolutionConfig
|
11
|
+
from ultralytics.utils import ASSETS_URL, LOGGER
|
11
12
|
from ultralytics.utils.checks import check_imshow, check_requirements
|
12
13
|
from ultralytics.utils.plotting import Annotator
|
13
14
|
|
@@ -72,15 +73,11 @@ class BaseSolution:
|
|
72
73
|
self.r_s = None
|
73
74
|
|
74
75
|
self.LOGGER = LOGGER # Store logger object to be used in multiple solution classes
|
75
|
-
|
76
|
-
|
77
|
-
DEFAULT_SOL_DICT.update(kwargs)
|
78
|
-
DEFAULT_CFG_DICT.update(kwargs)
|
79
|
-
self.CFG = {**DEFAULT_SOL_DICT, **DEFAULT_CFG_DICT}
|
80
|
-
self.LOGGER.info(f"Ultralytics Solutions: ✅ {DEFAULT_SOL_DICT}")
|
76
|
+
self.CFG = vars(SolutionConfig().update(**kwargs))
|
77
|
+
self.LOGGER.info(f"Ultralytics Solutions: ✅ {self.CFG}")
|
81
78
|
|
82
79
|
self.region = self.CFG["region"] # Store region data for other classes usage
|
83
|
-
self.line_width = self.CFG["line_width"]
|
80
|
+
self.line_width = self.CFG["line_width"]
|
84
81
|
|
85
82
|
# Load Model and store additional information (classes, show_conf, show_label)
|
86
83
|
if self.CFG["model"] is None:
|
@@ -178,7 +175,7 @@ class BaseSolution:
|
|
178
175
|
def initialize_region(self):
|
179
176
|
"""Initialize the counting region and line segment based on configuration settings."""
|
180
177
|
if self.region is None:
|
181
|
-
self.region = [(
|
178
|
+
self.region = [(10, 200), (540, 200), (540, 180), (10, 180)]
|
182
179
|
self.r_s = (
|
183
180
|
self.Polygon(self.region) if len(self.region) >= 3 else self.LineString(self.region)
|
184
181
|
) # region or line
|
@@ -44,15 +44,15 @@ class SpeedEstimator(BaseSolution):
|
|
44
44
|
"""
|
45
45
|
super().__init__(**kwargs)
|
46
46
|
|
47
|
-
self.fps =
|
47
|
+
self.fps = self.CFG["fps"] # assumed video FPS
|
48
48
|
self.frame_count = 0 # global frame count
|
49
49
|
self.trk_frame_ids = {} # Track ID → first frame index
|
50
50
|
self.spd = {} # Final speed per object (km/h), once locked
|
51
51
|
self.trk_hist = {} # Track ID → deque of (time, position)
|
52
52
|
self.locked_ids = set() # Track IDs whose speed has been finalized
|
53
|
-
self.max_hist =
|
54
|
-
self.meter_per_pixel =
|
55
|
-
self.max_speed =
|
53
|
+
self.max_hist = self.CFG["max_hist"] # Required frame history before computing speed
|
54
|
+
self.meter_per_pixel = self.CFG["meter_per_pixel"] # Scene scale, depends on camera details
|
55
|
+
self.max_speed = self.CFG["max_speed"] # max_speed adjustment
|
56
56
|
|
57
57
|
def process(self, im0):
|
58
58
|
"""
|
@@ -42,7 +42,7 @@ class TrackZone(BaseSolution):
|
|
42
42
|
**kwargs (Any): Additional keyword arguments passed to the parent class.
|
43
43
|
"""
|
44
44
|
super().__init__(**kwargs)
|
45
|
-
default_region = [(
|
45
|
+
default_region = [(75, 75), (565, 75), (565, 285), (75, 285)]
|
46
46
|
self.region = cv2.convexHull(np.array(self.region or default_region, dtype=np.int32))
|
47
47
|
|
48
48
|
def process(self, im0):
|
@@ -34,7 +34,7 @@ class VisionEye(BaseSolution):
|
|
34
34
|
"""
|
35
35
|
super().__init__(**kwargs)
|
36
36
|
# Set the vision point where the system will view objects and draw tracks
|
37
|
-
self.vision_point =
|
37
|
+
self.vision_point = self.CFG["vision_point"]
|
38
38
|
|
39
39
|
def process(self, im0):
|
40
40
|
"""
|
@@ -12,7 +12,6 @@ import subprocess
|
|
12
12
|
import sys
|
13
13
|
import threading
|
14
14
|
import time
|
15
|
-
import uuid
|
16
15
|
import warnings
|
17
16
|
from pathlib import Path
|
18
17
|
from threading import Lock
|
@@ -41,7 +40,6 @@ ROOT = FILE.parents[1] # YOLO
|
|
41
40
|
ASSETS = ROOT / "assets" # default images
|
42
41
|
ASSETS_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0" # assets GitHub URL
|
43
42
|
DEFAULT_CFG_PATH = ROOT / "cfg/default.yaml"
|
44
|
-
DEFAULT_SOL_CFG_PATH = ROOT / "cfg/solutions/default.yaml" # Ultralytics solutions yaml path
|
45
43
|
NUM_THREADS = min(8, max(1, os.cpu_count() - 1)) # number of YOLO multiprocessing threads
|
46
44
|
AUTOINSTALL = str(os.getenv("YOLO_AUTOINSTALL", True)).lower() == "true" # global auto-install mode
|
47
45
|
VERBOSE = str(os.getenv("YOLO_VERBOSE", True)).lower() == "true" # global verbose mode
|
@@ -545,7 +543,6 @@ def yaml_print(yaml_file: Union[str, Path, dict]) -> None:
|
|
545
543
|
|
546
544
|
# Default configuration
|
547
545
|
DEFAULT_CFG_DICT = yaml_load(DEFAULT_CFG_PATH)
|
548
|
-
DEFAULT_SOL_DICT = yaml_load(DEFAULT_SOL_CFG_PATH) # Ultralytics solutions configuration
|
549
546
|
for k, v in DEFAULT_CFG_DICT.items():
|
550
547
|
if isinstance(v, str) and v.lower() == "none":
|
551
548
|
DEFAULT_CFG_DICT[k] = None
|
@@ -1226,6 +1223,7 @@ class SettingsManager(JSONDict):
|
|
1226
1223
|
def __init__(self, file=SETTINGS_FILE, version="0.0.6"):
|
1227
1224
|
"""Initializes the SettingsManager with default settings and loads user settings."""
|
1228
1225
|
import hashlib
|
1226
|
+
import uuid
|
1229
1227
|
|
1230
1228
|
from ultralytics.utils.torch_utils import torch_distributed_zero_first
|
1231
1229
|
|
@@ -16,7 +16,6 @@ from typing import Optional
|
|
16
16
|
|
17
17
|
import cv2
|
18
18
|
import numpy as np
|
19
|
-
import requests
|
20
19
|
import torch
|
21
20
|
|
22
21
|
from ultralytics.utils import (
|
@@ -261,6 +260,8 @@ def check_latest_pypi_version(package_name="ultralytics"):
|
|
261
260
|
Returns:
|
262
261
|
(str): The latest version of the package.
|
263
262
|
"""
|
263
|
+
import requests # slow import
|
264
|
+
|
264
265
|
try:
|
265
266
|
requests.packages.urllib3.disable_warnings() # Disable the InsecureRequestWarning
|
266
267
|
response = requests.get(f"https://pypi.org/pypi/{package_name}/json", timeout=3)
|
@@ -2,7 +2,6 @@
|
|
2
2
|
|
3
3
|
import os
|
4
4
|
import shutil
|
5
|
-
import socket
|
6
5
|
import sys
|
7
6
|
import tempfile
|
8
7
|
|
@@ -20,6 +19,8 @@ def find_free_network_port() -> int:
|
|
20
19
|
Returns:
|
21
20
|
(int): The available network port number.
|
22
21
|
"""
|
22
|
+
import socket
|
23
|
+
|
23
24
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
24
25
|
s.bind(("127.0.0.1", 0))
|
25
26
|
return s.getsockname()[1] # port
|
@@ -8,7 +8,6 @@ from multiprocessing.pool import ThreadPool
|
|
8
8
|
from pathlib import Path
|
9
9
|
from urllib import parse, request
|
10
10
|
|
11
|
-
import requests
|
12
11
|
import torch
|
13
12
|
|
14
13
|
from ultralytics.utils import LOGGER, TQDM, checks, clean_url, emojis, is_online, url2file
|
@@ -203,6 +202,8 @@ def check_disk_space(url="https://ultralytics.com/assets/coco8.zip", path=Path.c
|
|
203
202
|
Returns:
|
204
203
|
(bool): True if there is sufficient disk space, False otherwise.
|
205
204
|
"""
|
205
|
+
import requests # slow import
|
206
|
+
|
206
207
|
try:
|
207
208
|
r = requests.head(url) # response
|
208
209
|
assert r.status_code < 400, f"URL error for {url}: {r.status_code} {r.reason}" # check response
|
@@ -244,6 +245,8 @@ def get_google_drive_file_info(link):
|
|
244
245
|
>>> link = "https://drive.google.com/file/d/1cqT-cJgANNrhIHCrEufUYhQ4RqiWG_lJ/view?usp=drive_link"
|
245
246
|
>>> url, filename = get_google_drive_file_info(link)
|
246
247
|
"""
|
248
|
+
import requests # slow import
|
249
|
+
|
247
250
|
file_id = link.split("/d/")[1].split("/view")[0]
|
248
251
|
drive_url = f"https://drive.google.com/uc?export=download&id={file_id}"
|
249
252
|
filename = None
|
@@ -388,6 +391,8 @@ def get_github_assets(repo="ultralytics/assets", version="latest", retry=False):
|
|
388
391
|
Examples:
|
389
392
|
>>> tag, assets = get_github_assets(repo="ultralytics/assets", version="latest")
|
390
393
|
"""
|
394
|
+
import requests # slow import
|
395
|
+
|
391
396
|
if version != "latest":
|
392
397
|
version = f"tags/{version}" # i.e. tags/v6.2
|
393
398
|
url = f"https://api.github.com/repos/{repo}/releases/{version}"
|
@@ -30,11 +30,6 @@ from ultralytics.utils import (
|
|
30
30
|
)
|
31
31
|
from ultralytics.utils.checks import check_version
|
32
32
|
|
33
|
-
try:
|
34
|
-
import thop
|
35
|
-
except ImportError:
|
36
|
-
thop = None # conda support without 'ultralytics-thop' installed
|
37
|
-
|
38
33
|
# Version checks (all default to version>=min_version)
|
39
34
|
TORCH_1_9 = check_version(torch.__version__, "1.9.0")
|
40
35
|
TORCH_1_13 = check_version(torch.__version__, "1.13.0")
|
@@ -404,6 +399,11 @@ def get_flops(model, imgsz=640):
|
|
404
399
|
Returns:
|
405
400
|
(float): The model FLOPs in billions.
|
406
401
|
"""
|
402
|
+
try:
|
403
|
+
import thop
|
404
|
+
except ImportError:
|
405
|
+
thop = None # conda support without 'ultralytics-thop' installed
|
406
|
+
|
407
407
|
if not thop:
|
408
408
|
return 0.0 # if not installed return 0.0 GFLOPs
|
409
409
|
|
@@ -811,6 +811,11 @@ def profile_ops(input, ops, n=10, device=None, max_num_obj=0):
|
|
811
811
|
>>> m2 = nn.SiLU()
|
812
812
|
>>> profile_ops(input, [m1, m2], n=100) # profile over 100 iterations
|
813
813
|
"""
|
814
|
+
try:
|
815
|
+
import thop
|
816
|
+
except ImportError:
|
817
|
+
thop = None # conda support without 'ultralytics-thop' installed
|
818
|
+
|
814
819
|
results = []
|
815
820
|
if not isinstance(device, torch.device):
|
816
821
|
device = select_device(device)
|