dgenerate-ultralytics-headless 8.3.253__py3-none-any.whl
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.
- dgenerate_ultralytics_headless-8.3.253.dist-info/METADATA +405 -0
- dgenerate_ultralytics_headless-8.3.253.dist-info/RECORD +299 -0
- dgenerate_ultralytics_headless-8.3.253.dist-info/WHEEL +5 -0
- dgenerate_ultralytics_headless-8.3.253.dist-info/entry_points.txt +3 -0
- dgenerate_ultralytics_headless-8.3.253.dist-info/licenses/LICENSE +661 -0
- dgenerate_ultralytics_headless-8.3.253.dist-info/top_level.txt +1 -0
- tests/__init__.py +23 -0
- tests/conftest.py +59 -0
- tests/test_cli.py +131 -0
- tests/test_cuda.py +216 -0
- tests/test_engine.py +157 -0
- tests/test_exports.py +309 -0
- tests/test_integrations.py +151 -0
- tests/test_python.py +777 -0
- tests/test_solutions.py +371 -0
- ultralytics/__init__.py +48 -0
- ultralytics/assets/bus.jpg +0 -0
- ultralytics/assets/zidane.jpg +0 -0
- ultralytics/cfg/__init__.py +1028 -0
- ultralytics/cfg/datasets/Argoverse.yaml +78 -0
- ultralytics/cfg/datasets/DOTAv1.5.yaml +37 -0
- ultralytics/cfg/datasets/DOTAv1.yaml +36 -0
- ultralytics/cfg/datasets/GlobalWheat2020.yaml +68 -0
- ultralytics/cfg/datasets/HomeObjects-3K.yaml +32 -0
- ultralytics/cfg/datasets/ImageNet.yaml +2025 -0
- ultralytics/cfg/datasets/Objects365.yaml +447 -0
- ultralytics/cfg/datasets/SKU-110K.yaml +58 -0
- ultralytics/cfg/datasets/TT100K.yaml +346 -0
- ultralytics/cfg/datasets/VOC.yaml +102 -0
- ultralytics/cfg/datasets/VisDrone.yaml +87 -0
- ultralytics/cfg/datasets/african-wildlife.yaml +25 -0
- ultralytics/cfg/datasets/brain-tumor.yaml +22 -0
- ultralytics/cfg/datasets/carparts-seg.yaml +44 -0
- ultralytics/cfg/datasets/coco-pose.yaml +64 -0
- ultralytics/cfg/datasets/coco.yaml +118 -0
- ultralytics/cfg/datasets/coco128-seg.yaml +101 -0
- ultralytics/cfg/datasets/coco128.yaml +101 -0
- ultralytics/cfg/datasets/coco8-grayscale.yaml +103 -0
- ultralytics/cfg/datasets/coco8-multispectral.yaml +104 -0
- ultralytics/cfg/datasets/coco8-pose.yaml +47 -0
- ultralytics/cfg/datasets/coco8-seg.yaml +101 -0
- ultralytics/cfg/datasets/coco8.yaml +101 -0
- ultralytics/cfg/datasets/construction-ppe.yaml +32 -0
- ultralytics/cfg/datasets/crack-seg.yaml +22 -0
- ultralytics/cfg/datasets/dog-pose.yaml +52 -0
- ultralytics/cfg/datasets/dota8-multispectral.yaml +38 -0
- ultralytics/cfg/datasets/dota8.yaml +35 -0
- ultralytics/cfg/datasets/hand-keypoints.yaml +50 -0
- ultralytics/cfg/datasets/kitti.yaml +27 -0
- ultralytics/cfg/datasets/lvis.yaml +1240 -0
- ultralytics/cfg/datasets/medical-pills.yaml +21 -0
- ultralytics/cfg/datasets/open-images-v7.yaml +663 -0
- ultralytics/cfg/datasets/package-seg.yaml +22 -0
- ultralytics/cfg/datasets/signature.yaml +21 -0
- ultralytics/cfg/datasets/tiger-pose.yaml +41 -0
- ultralytics/cfg/datasets/xView.yaml +155 -0
- ultralytics/cfg/default.yaml +130 -0
- ultralytics/cfg/models/11/yolo11-cls-resnet18.yaml +17 -0
- ultralytics/cfg/models/11/yolo11-cls.yaml +33 -0
- ultralytics/cfg/models/11/yolo11-obb.yaml +50 -0
- ultralytics/cfg/models/11/yolo11-pose.yaml +51 -0
- ultralytics/cfg/models/11/yolo11-seg.yaml +50 -0
- ultralytics/cfg/models/11/yolo11.yaml +50 -0
- ultralytics/cfg/models/11/yoloe-11-seg.yaml +48 -0
- ultralytics/cfg/models/11/yoloe-11.yaml +48 -0
- ultralytics/cfg/models/12/yolo12-cls.yaml +32 -0
- ultralytics/cfg/models/12/yolo12-obb.yaml +48 -0
- ultralytics/cfg/models/12/yolo12-pose.yaml +49 -0
- ultralytics/cfg/models/12/yolo12-seg.yaml +48 -0
- ultralytics/cfg/models/12/yolo12.yaml +48 -0
- ultralytics/cfg/models/rt-detr/rtdetr-l.yaml +53 -0
- ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml +45 -0
- ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml +45 -0
- ultralytics/cfg/models/rt-detr/rtdetr-x.yaml +57 -0
- ultralytics/cfg/models/v10/yolov10b.yaml +45 -0
- ultralytics/cfg/models/v10/yolov10l.yaml +45 -0
- ultralytics/cfg/models/v10/yolov10m.yaml +45 -0
- ultralytics/cfg/models/v10/yolov10n.yaml +45 -0
- ultralytics/cfg/models/v10/yolov10s.yaml +45 -0
- ultralytics/cfg/models/v10/yolov10x.yaml +45 -0
- ultralytics/cfg/models/v3/yolov3-spp.yaml +49 -0
- ultralytics/cfg/models/v3/yolov3-tiny.yaml +40 -0
- ultralytics/cfg/models/v3/yolov3.yaml +49 -0
- ultralytics/cfg/models/v5/yolov5-p6.yaml +62 -0
- ultralytics/cfg/models/v5/yolov5.yaml +51 -0
- ultralytics/cfg/models/v6/yolov6.yaml +56 -0
- ultralytics/cfg/models/v8/yoloe-v8-seg.yaml +48 -0
- ultralytics/cfg/models/v8/yoloe-v8.yaml +48 -0
- ultralytics/cfg/models/v8/yolov8-cls-resnet101.yaml +28 -0
- ultralytics/cfg/models/v8/yolov8-cls-resnet50.yaml +28 -0
- ultralytics/cfg/models/v8/yolov8-cls.yaml +32 -0
- ultralytics/cfg/models/v8/yolov8-ghost-p2.yaml +58 -0
- ultralytics/cfg/models/v8/yolov8-ghost-p6.yaml +60 -0
- ultralytics/cfg/models/v8/yolov8-ghost.yaml +50 -0
- ultralytics/cfg/models/v8/yolov8-obb.yaml +49 -0
- ultralytics/cfg/models/v8/yolov8-p2.yaml +57 -0
- ultralytics/cfg/models/v8/yolov8-p6.yaml +59 -0
- ultralytics/cfg/models/v8/yolov8-pose-p6.yaml +60 -0
- ultralytics/cfg/models/v8/yolov8-pose.yaml +50 -0
- ultralytics/cfg/models/v8/yolov8-rtdetr.yaml +49 -0
- ultralytics/cfg/models/v8/yolov8-seg-p6.yaml +59 -0
- ultralytics/cfg/models/v8/yolov8-seg.yaml +49 -0
- ultralytics/cfg/models/v8/yolov8-world.yaml +51 -0
- ultralytics/cfg/models/v8/yolov8-worldv2.yaml +49 -0
- ultralytics/cfg/models/v8/yolov8.yaml +49 -0
- ultralytics/cfg/models/v9/yolov9c-seg.yaml +41 -0
- ultralytics/cfg/models/v9/yolov9c.yaml +41 -0
- ultralytics/cfg/models/v9/yolov9e-seg.yaml +64 -0
- ultralytics/cfg/models/v9/yolov9e.yaml +64 -0
- ultralytics/cfg/models/v9/yolov9m.yaml +41 -0
- ultralytics/cfg/models/v9/yolov9s.yaml +41 -0
- ultralytics/cfg/models/v9/yolov9t.yaml +41 -0
- ultralytics/cfg/trackers/botsort.yaml +21 -0
- ultralytics/cfg/trackers/bytetrack.yaml +12 -0
- ultralytics/data/__init__.py +26 -0
- ultralytics/data/annotator.py +66 -0
- ultralytics/data/augment.py +2801 -0
- ultralytics/data/base.py +435 -0
- ultralytics/data/build.py +437 -0
- ultralytics/data/converter.py +855 -0
- ultralytics/data/dataset.py +834 -0
- ultralytics/data/loaders.py +704 -0
- ultralytics/data/scripts/download_weights.sh +18 -0
- ultralytics/data/scripts/get_coco.sh +61 -0
- ultralytics/data/scripts/get_coco128.sh +18 -0
- ultralytics/data/scripts/get_imagenet.sh +52 -0
- ultralytics/data/split.py +138 -0
- ultralytics/data/split_dota.py +344 -0
- ultralytics/data/utils.py +798 -0
- ultralytics/engine/__init__.py +1 -0
- ultralytics/engine/exporter.py +1580 -0
- ultralytics/engine/model.py +1125 -0
- ultralytics/engine/predictor.py +508 -0
- ultralytics/engine/results.py +1522 -0
- ultralytics/engine/trainer.py +977 -0
- ultralytics/engine/tuner.py +449 -0
- ultralytics/engine/validator.py +387 -0
- ultralytics/hub/__init__.py +166 -0
- ultralytics/hub/auth.py +151 -0
- ultralytics/hub/google/__init__.py +174 -0
- ultralytics/hub/session.py +422 -0
- ultralytics/hub/utils.py +162 -0
- ultralytics/models/__init__.py +9 -0
- ultralytics/models/fastsam/__init__.py +7 -0
- ultralytics/models/fastsam/model.py +79 -0
- ultralytics/models/fastsam/predict.py +169 -0
- ultralytics/models/fastsam/utils.py +23 -0
- ultralytics/models/fastsam/val.py +38 -0
- ultralytics/models/nas/__init__.py +7 -0
- ultralytics/models/nas/model.py +98 -0
- ultralytics/models/nas/predict.py +56 -0
- ultralytics/models/nas/val.py +38 -0
- ultralytics/models/rtdetr/__init__.py +7 -0
- ultralytics/models/rtdetr/model.py +63 -0
- ultralytics/models/rtdetr/predict.py +88 -0
- ultralytics/models/rtdetr/train.py +89 -0
- ultralytics/models/rtdetr/val.py +216 -0
- ultralytics/models/sam/__init__.py +25 -0
- ultralytics/models/sam/amg.py +275 -0
- ultralytics/models/sam/build.py +365 -0
- ultralytics/models/sam/build_sam3.py +377 -0
- ultralytics/models/sam/model.py +169 -0
- ultralytics/models/sam/modules/__init__.py +1 -0
- ultralytics/models/sam/modules/blocks.py +1067 -0
- ultralytics/models/sam/modules/decoders.py +495 -0
- ultralytics/models/sam/modules/encoders.py +794 -0
- ultralytics/models/sam/modules/memory_attention.py +298 -0
- ultralytics/models/sam/modules/sam.py +1160 -0
- ultralytics/models/sam/modules/tiny_encoder.py +979 -0
- ultralytics/models/sam/modules/transformer.py +344 -0
- ultralytics/models/sam/modules/utils.py +512 -0
- ultralytics/models/sam/predict.py +3940 -0
- ultralytics/models/sam/sam3/__init__.py +3 -0
- ultralytics/models/sam/sam3/decoder.py +546 -0
- ultralytics/models/sam/sam3/encoder.py +529 -0
- ultralytics/models/sam/sam3/geometry_encoders.py +415 -0
- ultralytics/models/sam/sam3/maskformer_segmentation.py +286 -0
- ultralytics/models/sam/sam3/model_misc.py +199 -0
- ultralytics/models/sam/sam3/necks.py +129 -0
- ultralytics/models/sam/sam3/sam3_image.py +339 -0
- ultralytics/models/sam/sam3/text_encoder_ve.py +307 -0
- ultralytics/models/sam/sam3/vitdet.py +547 -0
- ultralytics/models/sam/sam3/vl_combiner.py +160 -0
- ultralytics/models/utils/__init__.py +1 -0
- ultralytics/models/utils/loss.py +466 -0
- ultralytics/models/utils/ops.py +315 -0
- ultralytics/models/yolo/__init__.py +7 -0
- ultralytics/models/yolo/classify/__init__.py +7 -0
- ultralytics/models/yolo/classify/predict.py +90 -0
- ultralytics/models/yolo/classify/train.py +202 -0
- ultralytics/models/yolo/classify/val.py +216 -0
- ultralytics/models/yolo/detect/__init__.py +7 -0
- ultralytics/models/yolo/detect/predict.py +122 -0
- ultralytics/models/yolo/detect/train.py +227 -0
- ultralytics/models/yolo/detect/val.py +507 -0
- ultralytics/models/yolo/model.py +430 -0
- ultralytics/models/yolo/obb/__init__.py +7 -0
- ultralytics/models/yolo/obb/predict.py +56 -0
- ultralytics/models/yolo/obb/train.py +79 -0
- ultralytics/models/yolo/obb/val.py +302 -0
- ultralytics/models/yolo/pose/__init__.py +7 -0
- ultralytics/models/yolo/pose/predict.py +65 -0
- ultralytics/models/yolo/pose/train.py +110 -0
- ultralytics/models/yolo/pose/val.py +248 -0
- ultralytics/models/yolo/segment/__init__.py +7 -0
- ultralytics/models/yolo/segment/predict.py +109 -0
- ultralytics/models/yolo/segment/train.py +69 -0
- ultralytics/models/yolo/segment/val.py +307 -0
- ultralytics/models/yolo/world/__init__.py +5 -0
- ultralytics/models/yolo/world/train.py +173 -0
- ultralytics/models/yolo/world/train_world.py +178 -0
- ultralytics/models/yolo/yoloe/__init__.py +22 -0
- ultralytics/models/yolo/yoloe/predict.py +162 -0
- ultralytics/models/yolo/yoloe/train.py +287 -0
- ultralytics/models/yolo/yoloe/train_seg.py +122 -0
- ultralytics/models/yolo/yoloe/val.py +206 -0
- ultralytics/nn/__init__.py +27 -0
- ultralytics/nn/autobackend.py +964 -0
- ultralytics/nn/modules/__init__.py +182 -0
- ultralytics/nn/modules/activation.py +54 -0
- ultralytics/nn/modules/block.py +1947 -0
- ultralytics/nn/modules/conv.py +669 -0
- ultralytics/nn/modules/head.py +1183 -0
- ultralytics/nn/modules/transformer.py +793 -0
- ultralytics/nn/modules/utils.py +159 -0
- ultralytics/nn/tasks.py +1768 -0
- ultralytics/nn/text_model.py +356 -0
- ultralytics/py.typed +1 -0
- ultralytics/solutions/__init__.py +41 -0
- ultralytics/solutions/ai_gym.py +108 -0
- ultralytics/solutions/analytics.py +264 -0
- ultralytics/solutions/config.py +107 -0
- ultralytics/solutions/distance_calculation.py +123 -0
- ultralytics/solutions/heatmap.py +125 -0
- ultralytics/solutions/instance_segmentation.py +86 -0
- ultralytics/solutions/object_blurrer.py +89 -0
- ultralytics/solutions/object_counter.py +190 -0
- ultralytics/solutions/object_cropper.py +87 -0
- ultralytics/solutions/parking_management.py +280 -0
- ultralytics/solutions/queue_management.py +93 -0
- ultralytics/solutions/region_counter.py +133 -0
- ultralytics/solutions/security_alarm.py +151 -0
- ultralytics/solutions/similarity_search.py +219 -0
- ultralytics/solutions/solutions.py +828 -0
- ultralytics/solutions/speed_estimation.py +114 -0
- ultralytics/solutions/streamlit_inference.py +260 -0
- ultralytics/solutions/templates/similarity-search.html +156 -0
- ultralytics/solutions/trackzone.py +88 -0
- ultralytics/solutions/vision_eye.py +67 -0
- ultralytics/trackers/__init__.py +7 -0
- ultralytics/trackers/basetrack.py +115 -0
- ultralytics/trackers/bot_sort.py +257 -0
- ultralytics/trackers/byte_tracker.py +469 -0
- ultralytics/trackers/track.py +116 -0
- ultralytics/trackers/utils/__init__.py +1 -0
- ultralytics/trackers/utils/gmc.py +339 -0
- ultralytics/trackers/utils/kalman_filter.py +482 -0
- ultralytics/trackers/utils/matching.py +154 -0
- ultralytics/utils/__init__.py +1450 -0
- ultralytics/utils/autobatch.py +118 -0
- ultralytics/utils/autodevice.py +205 -0
- ultralytics/utils/benchmarks.py +728 -0
- ultralytics/utils/callbacks/__init__.py +5 -0
- ultralytics/utils/callbacks/base.py +233 -0
- ultralytics/utils/callbacks/clearml.py +146 -0
- ultralytics/utils/callbacks/comet.py +625 -0
- ultralytics/utils/callbacks/dvc.py +197 -0
- ultralytics/utils/callbacks/hub.py +110 -0
- ultralytics/utils/callbacks/mlflow.py +134 -0
- ultralytics/utils/callbacks/neptune.py +126 -0
- ultralytics/utils/callbacks/platform.py +453 -0
- ultralytics/utils/callbacks/raytune.py +42 -0
- ultralytics/utils/callbacks/tensorboard.py +123 -0
- ultralytics/utils/callbacks/wb.py +188 -0
- ultralytics/utils/checks.py +1020 -0
- ultralytics/utils/cpu.py +85 -0
- ultralytics/utils/dist.py +123 -0
- ultralytics/utils/downloads.py +529 -0
- ultralytics/utils/errors.py +35 -0
- ultralytics/utils/events.py +113 -0
- ultralytics/utils/export/__init__.py +7 -0
- ultralytics/utils/export/engine.py +237 -0
- ultralytics/utils/export/imx.py +325 -0
- ultralytics/utils/export/tensorflow.py +231 -0
- ultralytics/utils/files.py +219 -0
- ultralytics/utils/git.py +137 -0
- ultralytics/utils/instance.py +484 -0
- ultralytics/utils/logger.py +506 -0
- ultralytics/utils/loss.py +849 -0
- ultralytics/utils/metrics.py +1563 -0
- ultralytics/utils/nms.py +337 -0
- ultralytics/utils/ops.py +664 -0
- ultralytics/utils/patches.py +201 -0
- ultralytics/utils/plotting.py +1047 -0
- ultralytics/utils/tal.py +404 -0
- ultralytics/utils/torch_utils.py +984 -0
- ultralytics/utils/tqdm.py +443 -0
- ultralytics/utils/triton.py +112 -0
- ultralytics/utils/tuner.py +168 -0
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# COCO8 dataset (first 8 images from COCO train2017) by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/detect/coco8/
|
|
5
|
+
# Example usage: yolo train data=coco8.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── coco8 ← downloads here (1 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: coco8 # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 4 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 4 images
|
|
15
|
+
test: # test images (optional)
|
|
16
|
+
|
|
17
|
+
# Classes
|
|
18
|
+
names:
|
|
19
|
+
0: person
|
|
20
|
+
1: bicycle
|
|
21
|
+
2: car
|
|
22
|
+
3: motorcycle
|
|
23
|
+
4: airplane
|
|
24
|
+
5: bus
|
|
25
|
+
6: train
|
|
26
|
+
7: truck
|
|
27
|
+
8: boat
|
|
28
|
+
9: traffic light
|
|
29
|
+
10: fire hydrant
|
|
30
|
+
11: stop sign
|
|
31
|
+
12: parking meter
|
|
32
|
+
13: bench
|
|
33
|
+
14: bird
|
|
34
|
+
15: cat
|
|
35
|
+
16: dog
|
|
36
|
+
17: horse
|
|
37
|
+
18: sheep
|
|
38
|
+
19: cow
|
|
39
|
+
20: elephant
|
|
40
|
+
21: bear
|
|
41
|
+
22: zebra
|
|
42
|
+
23: giraffe
|
|
43
|
+
24: backpack
|
|
44
|
+
25: umbrella
|
|
45
|
+
26: handbag
|
|
46
|
+
27: tie
|
|
47
|
+
28: suitcase
|
|
48
|
+
29: frisbee
|
|
49
|
+
30: skis
|
|
50
|
+
31: snowboard
|
|
51
|
+
32: sports ball
|
|
52
|
+
33: kite
|
|
53
|
+
34: baseball bat
|
|
54
|
+
35: baseball glove
|
|
55
|
+
36: skateboard
|
|
56
|
+
37: surfboard
|
|
57
|
+
38: tennis racket
|
|
58
|
+
39: bottle
|
|
59
|
+
40: wine glass
|
|
60
|
+
41: cup
|
|
61
|
+
42: fork
|
|
62
|
+
43: knife
|
|
63
|
+
44: spoon
|
|
64
|
+
45: bowl
|
|
65
|
+
46: banana
|
|
66
|
+
47: apple
|
|
67
|
+
48: sandwich
|
|
68
|
+
49: orange
|
|
69
|
+
50: broccoli
|
|
70
|
+
51: carrot
|
|
71
|
+
52: hot dog
|
|
72
|
+
53: pizza
|
|
73
|
+
54: donut
|
|
74
|
+
55: cake
|
|
75
|
+
56: chair
|
|
76
|
+
57: couch
|
|
77
|
+
58: potted plant
|
|
78
|
+
59: bed
|
|
79
|
+
60: dining table
|
|
80
|
+
61: toilet
|
|
81
|
+
62: tv
|
|
82
|
+
63: laptop
|
|
83
|
+
64: mouse
|
|
84
|
+
65: remote
|
|
85
|
+
66: keyboard
|
|
86
|
+
67: cell phone
|
|
87
|
+
68: microwave
|
|
88
|
+
69: oven
|
|
89
|
+
70: toaster
|
|
90
|
+
71: sink
|
|
91
|
+
72: refrigerator
|
|
92
|
+
73: book
|
|
93
|
+
74: clock
|
|
94
|
+
75: vase
|
|
95
|
+
76: scissors
|
|
96
|
+
77: teddy bear
|
|
97
|
+
78: hair drier
|
|
98
|
+
79: toothbrush
|
|
99
|
+
|
|
100
|
+
# Download script/URL (optional)
|
|
101
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8.zip
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# Construction-PPE dataset by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/detect/construction-ppe/
|
|
5
|
+
# Example usage: yolo train data=construction-ppe.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── construction-ppe ← downloads here (178.4 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: construction-ppe # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 1132 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 143 images
|
|
15
|
+
test: images/test # test images (relative to 'path') 141 images
|
|
16
|
+
|
|
17
|
+
# Classes
|
|
18
|
+
names:
|
|
19
|
+
0: helmet
|
|
20
|
+
1: gloves
|
|
21
|
+
2: vest
|
|
22
|
+
3: boots
|
|
23
|
+
4: goggles
|
|
24
|
+
5: none
|
|
25
|
+
6: Person
|
|
26
|
+
7: no_helmet
|
|
27
|
+
8: no_goggle
|
|
28
|
+
9: no_gloves
|
|
29
|
+
10: no_boots
|
|
30
|
+
|
|
31
|
+
# Download script/URL (optional)
|
|
32
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/construction-ppe.zip
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# Crack-seg dataset by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/segment/crack-seg/
|
|
5
|
+
# Example usage: yolo train data=crack-seg.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── crack-seg ← downloads here (91.6 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: crack-seg # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 3717 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 112 images
|
|
15
|
+
test: images/test # test images (relative to 'path') 200 images
|
|
16
|
+
|
|
17
|
+
# Classes
|
|
18
|
+
names:
|
|
19
|
+
0: crack
|
|
20
|
+
|
|
21
|
+
# Download script/URL (optional)
|
|
22
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/crack-seg.zip
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# Dogs dataset http://vision.stanford.edu/aditya86/ImageNetDogs/ by Stanford
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/pose/dog-pose/
|
|
5
|
+
# Example usage: yolo train data=dog-pose.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── dog-pose ← downloads here (337 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: dog-pose # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 6773 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 1703 images
|
|
15
|
+
|
|
16
|
+
# Keypoints
|
|
17
|
+
kpt_shape: [24, 3] # number of keypoints, number of dims (2 for x,y or 3 for x,y,visible)
|
|
18
|
+
|
|
19
|
+
# Classes
|
|
20
|
+
names:
|
|
21
|
+
0: dog
|
|
22
|
+
|
|
23
|
+
# Keypoint names per class
|
|
24
|
+
kpt_names:
|
|
25
|
+
0:
|
|
26
|
+
- front_left_paw
|
|
27
|
+
- front_left_knee
|
|
28
|
+
- front_left_elbow
|
|
29
|
+
- rear_left_paw
|
|
30
|
+
- rear_left_knee
|
|
31
|
+
- rear_left_elbow
|
|
32
|
+
- front_right_paw
|
|
33
|
+
- front_right_knee
|
|
34
|
+
- front_right_elbow
|
|
35
|
+
- rear_right_paw
|
|
36
|
+
- rear_right_knee
|
|
37
|
+
- rear_right_elbow
|
|
38
|
+
- tail_start
|
|
39
|
+
- tail_end
|
|
40
|
+
- left_ear_base
|
|
41
|
+
- right_ear_base
|
|
42
|
+
- nose
|
|
43
|
+
- chin
|
|
44
|
+
- left_ear_tip
|
|
45
|
+
- right_ear_tip
|
|
46
|
+
- left_eye
|
|
47
|
+
- right_eye
|
|
48
|
+
- withers
|
|
49
|
+
- throat
|
|
50
|
+
|
|
51
|
+
# Download script/URL (optional)
|
|
52
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/dog-pose.zip
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# DOTA8-Multispectral dataset (DOTA8 interpolated across 10 channels in the visual spectrum) by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/obb/dota8/
|
|
5
|
+
# Example usage: yolo train model=yolov8n-obb.pt data=dota8-multispectral.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── dota8-multispectral ← downloads here (37.3 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: dota8-multispectral # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 4 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 4 images
|
|
15
|
+
|
|
16
|
+
# Number of multispectral image channels
|
|
17
|
+
channels: 10
|
|
18
|
+
|
|
19
|
+
# Classes for DOTA 1.0
|
|
20
|
+
names:
|
|
21
|
+
0: plane
|
|
22
|
+
1: ship
|
|
23
|
+
2: storage tank
|
|
24
|
+
3: baseball diamond
|
|
25
|
+
4: tennis court
|
|
26
|
+
5: basketball court
|
|
27
|
+
6: ground track field
|
|
28
|
+
7: harbor
|
|
29
|
+
8: bridge
|
|
30
|
+
9: large vehicle
|
|
31
|
+
10: small vehicle
|
|
32
|
+
11: helicopter
|
|
33
|
+
12: roundabout
|
|
34
|
+
13: soccer ball field
|
|
35
|
+
14: swimming pool
|
|
36
|
+
|
|
37
|
+
# Download script/URL (optional)
|
|
38
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/dota8-multispectral.zip
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# DOTA8 dataset (8 images from the DOTAv1 split) by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/obb/dota8/
|
|
5
|
+
# Example usage: yolo train model=yolov8n-obb.pt data=dota8.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── dota8 ← downloads here (1 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: dota8 # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 4 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 4 images
|
|
15
|
+
|
|
16
|
+
# Classes for DOTA 1.0
|
|
17
|
+
names:
|
|
18
|
+
0: plane
|
|
19
|
+
1: ship
|
|
20
|
+
2: storage tank
|
|
21
|
+
3: baseball diamond
|
|
22
|
+
4: tennis court
|
|
23
|
+
5: basketball court
|
|
24
|
+
6: ground track field
|
|
25
|
+
7: harbor
|
|
26
|
+
8: bridge
|
|
27
|
+
9: large vehicle
|
|
28
|
+
10: small vehicle
|
|
29
|
+
11: helicopter
|
|
30
|
+
12: roundabout
|
|
31
|
+
13: soccer ball field
|
|
32
|
+
14: swimming pool
|
|
33
|
+
|
|
34
|
+
# Download script/URL (optional)
|
|
35
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/dota8.zip
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# Hand Keypoints dataset by Ultralytics
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/pose/hand-keypoints/
|
|
5
|
+
# Example usage: yolo train data=hand-keypoints.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── hand-keypoints ← downloads here (369 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: hand-keypoints # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 18776 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 7992 images
|
|
15
|
+
|
|
16
|
+
# Keypoints
|
|
17
|
+
kpt_shape: [21, 3] # number of keypoints, number of dims (2 for x,y or 3 for x,y,visible)
|
|
18
|
+
flip_idx: [0, 1, 2, 4, 3, 10, 11, 12, 13, 14, 5, 6, 7, 8, 9, 15, 16, 17, 18, 19, 20]
|
|
19
|
+
|
|
20
|
+
# Classes
|
|
21
|
+
names:
|
|
22
|
+
0: hand
|
|
23
|
+
|
|
24
|
+
# Keypoint names per class
|
|
25
|
+
kpt_names:
|
|
26
|
+
0:
|
|
27
|
+
- wrist
|
|
28
|
+
- thumb_cmc
|
|
29
|
+
- thumb_mcp
|
|
30
|
+
- thumb_ip
|
|
31
|
+
- thumb_tip
|
|
32
|
+
- index_mcp
|
|
33
|
+
- index_pip
|
|
34
|
+
- index_dip
|
|
35
|
+
- index_tip
|
|
36
|
+
- middle_mcp
|
|
37
|
+
- middle_pip
|
|
38
|
+
- middle_dip
|
|
39
|
+
- middle_tip
|
|
40
|
+
- ring_mcp
|
|
41
|
+
- ring_pip
|
|
42
|
+
- ring_dip
|
|
43
|
+
- ring_tip
|
|
44
|
+
- pinky_mcp
|
|
45
|
+
- pinky_pip
|
|
46
|
+
- pinky_dip
|
|
47
|
+
- pinky_tip
|
|
48
|
+
|
|
49
|
+
# Download script/URL (optional)
|
|
50
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/hand-keypoints.zip
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
|
+
|
|
3
|
+
# KITTI dataset by Karlsruhe Institute of Technology and Toyota Technological Institute at Chicago
|
|
4
|
+
# Documentation: https://docs.ultralytics.com/datasets/detect/kitti/
|
|
5
|
+
# Example usage: yolo train data=kitti.yaml
|
|
6
|
+
# parent
|
|
7
|
+
# ├── ultralytics
|
|
8
|
+
# └── datasets
|
|
9
|
+
# └── kitti ← downloads here (390.5 MB)
|
|
10
|
+
|
|
11
|
+
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
|
|
12
|
+
path: kitti # dataset root dir
|
|
13
|
+
train: images/train # train images (relative to 'path') 5985 images
|
|
14
|
+
val: images/val # val images (relative to 'path') 1496 images
|
|
15
|
+
|
|
16
|
+
names:
|
|
17
|
+
0: car
|
|
18
|
+
1: van
|
|
19
|
+
2: truck
|
|
20
|
+
3: pedestrian
|
|
21
|
+
4: person_sitting
|
|
22
|
+
5: cyclist
|
|
23
|
+
6: tram
|
|
24
|
+
7: misc
|
|
25
|
+
|
|
26
|
+
# Download script/URL (optional)
|
|
27
|
+
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/kitti.zip
|