mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
3ec95f9e3d
* Add license line to .github/ISSUE_TEMPLATE/bug-report.yml * Add license line to .github/ISSUE_TEMPLATE/config.yml * Add license line to .github/ISSUE_TEMPLATE/feature-request.yml * Add license line to .github/ISSUE_TEMPLATE/question.yml * Add license line to .github/dependabot.yml * Add license line to .github/workflows/ci-testing.yml * Add license line to .github/workflows/cla.yml * Add license line to .github/workflows/codeql-analysis.yml * Add license line to .github/workflows/docker.yml * Add license line to .github/workflows/format.yml * Add license line to .github/workflows/greetings.yml * Add license line to .github/workflows/links.yml * Add license line to .github/workflows/merge-main-into-prs.yml * Add license line to .github/workflows/stale.yml * Add license line to benchmarks.py * Add license line to classify/predict.py * Add license line to classify/train.py * Add license line to classify/val.py * Add license line to data/Argoverse.yaml * Add license line to data/GlobalWheat2020.yaml * Add license line to data/ImageNet.yaml * Add license line to data/ImageNet10.yaml * Add license line to data/ImageNet100.yaml * Add license line to data/ImageNet1000.yaml * Add license line to data/Objects365.yaml * Add license line to data/SKU-110K.yaml * Add license line to data/VOC.yaml * Add license line to data/VisDrone.yaml * Add license line to data/coco.yaml * Add license line to data/coco128-seg.yaml * Add license line to data/coco128.yaml * Add license line to data/hyps/hyp.Objects365.yaml * Add license line to data/hyps/hyp.VOC.yaml * Add license line to data/hyps/hyp.no-augmentation.yaml * Add license line to data/hyps/hyp.scratch-high.yaml * Add license line to data/hyps/hyp.scratch-low.yaml * Add license line to data/hyps/hyp.scratch-med.yaml * Add license line to data/xView.yaml * Add license line to detect.py * Add license line to export.py * Add license line to hubconf.py * Add license line to models/common.py * Add license line to models/experimental.py * Add license line to models/hub/anchors.yaml * Add license line to models/hub/yolov3-spp.yaml * Add license line to models/hub/yolov3-tiny.yaml * Add license line to models/hub/yolov3.yaml * Add license line to models/hub/yolov5-bifpn.yaml * Add license line to models/hub/yolov5-fpn.yaml * Add license line to models/hub/yolov5-p2.yaml * Add license line to models/hub/yolov5-p34.yaml * Add license line to models/hub/yolov5-p6.yaml * Add license line to models/hub/yolov5-p7.yaml * Add license line to models/hub/yolov5-panet.yaml * Add license line to models/hub/yolov5l6.yaml * Add license line to models/hub/yolov5m6.yaml * Add license line to models/hub/yolov5n6.yaml * Add license line to models/hub/yolov5s-LeakyReLU.yaml * Add license line to models/hub/yolov5s-ghost.yaml * Add license line to models/hub/yolov5s-transformer.yaml * Add license line to models/hub/yolov5s6.yaml * Add license line to models/hub/yolov5x6.yaml * Add license line to models/segment/yolov5l-seg.yaml * Add license line to models/segment/yolov5m-seg.yaml * Add license line to models/segment/yolov5n-seg.yaml * Add license line to models/segment/yolov5s-seg.yaml * Add license line to models/segment/yolov5x-seg.yaml * Add license line to models/tf.py * Add license line to models/yolo.py * Add license line to models/yolov5l.yaml * Add license line to models/yolov5m.yaml * Add license line to models/yolov5n.yaml * Add license line to models/yolov5s.yaml * Add license line to models/yolov5x.yaml * Add license line to pyproject.toml * Add license line to segment/predict.py * Add license line to segment/train.py * Add license line to segment/val.py * Add license line to train.py * Add license line to utils/__init__.py * Add license line to utils/activations.py * Add license line to utils/augmentations.py * Add license line to utils/autoanchor.py * Add license line to utils/autobatch.py * Add license line to utils/aws/resume.py * Add license line to utils/callbacks.py * Add license line to utils/dataloaders.py * Add license line to utils/downloads.py * Add license line to utils/flask_rest_api/example_request.py * Add license line to utils/flask_rest_api/restapi.py * Add license line to utils/general.py * Add license line to utils/google_app_engine/app.yaml * Add license line to utils/loggers/__init__.py * Add license line to utils/loggers/clearml/clearml_utils.py * Add license line to utils/loggers/clearml/hpo.py * Add license line to utils/loggers/comet/__init__.py * Add license line to utils/loggers/comet/comet_utils.py * Add license line to utils/loggers/comet/hpo.py * Add license line to utils/loggers/wandb/wandb_utils.py * Add license line to utils/loss.py * Add license line to utils/metrics.py * Add license line to utils/plots.py * Add license line to utils/segment/augmentations.py * Add license line to utils/segment/dataloaders.py * Add license line to utils/segment/general.py * Add license line to utils/segment/loss.py * Add license line to utils/segment/metrics.py * Add license line to utils/segment/plots.py * Add license line to utils/torch_utils.py * Add license line to utils/triton.py * Add license line to val.py * Auto-format by https://ultralytics.com/actions * Update ImageNet1000.yaml Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Auto-format by https://ultralytics.com/actions --------- Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
115 lines
2.4 KiB
YAML
115 lines
2.4 KiB
YAML
# Ultralytics YOLOv5 🚀, AGPL-3.0 license
|
|
# COCO 2017 dataset http://cocodataset.org by Microsoft
|
|
# Example usage: python train.py --data coco.yaml
|
|
# parent
|
|
# ├── yolov5
|
|
# └── datasets
|
|
# └── coco ← downloads here (20.1 GB)
|
|
|
|
# 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, ..]
|
|
path: ../datasets/coco # dataset root dir
|
|
train: train2017.txt # train images (relative to 'path') 118287 images
|
|
val: val2017.txt # val images (relative to 'path') 5000 images
|
|
test: test-dev2017.txt # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794
|
|
|
|
# Classes
|
|
names:
|
|
0: person
|
|
1: bicycle
|
|
2: car
|
|
3: motorcycle
|
|
4: airplane
|
|
5: bus
|
|
6: train
|
|
7: truck
|
|
8: boat
|
|
9: traffic light
|
|
10: fire hydrant
|
|
11: stop sign
|
|
12: parking meter
|
|
13: bench
|
|
14: bird
|
|
15: cat
|
|
16: dog
|
|
17: horse
|
|
18: sheep
|
|
19: cow
|
|
20: elephant
|
|
21: bear
|
|
22: zebra
|
|
23: giraffe
|
|
24: backpack
|
|
25: umbrella
|
|
26: handbag
|
|
27: tie
|
|
28: suitcase
|
|
29: frisbee
|
|
30: skis
|
|
31: snowboard
|
|
32: sports ball
|
|
33: kite
|
|
34: baseball bat
|
|
35: baseball glove
|
|
36: skateboard
|
|
37: surfboard
|
|
38: tennis racket
|
|
39: bottle
|
|
40: wine glass
|
|
41: cup
|
|
42: fork
|
|
43: knife
|
|
44: spoon
|
|
45: bowl
|
|
46: banana
|
|
47: apple
|
|
48: sandwich
|
|
49: orange
|
|
50: broccoli
|
|
51: carrot
|
|
52: hot dog
|
|
53: pizza
|
|
54: donut
|
|
55: cake
|
|
56: chair
|
|
57: couch
|
|
58: potted plant
|
|
59: bed
|
|
60: dining table
|
|
61: toilet
|
|
62: tv
|
|
63: laptop
|
|
64: mouse
|
|
65: remote
|
|
66: keyboard
|
|
67: cell phone
|
|
68: microwave
|
|
69: oven
|
|
70: toaster
|
|
71: sink
|
|
72: refrigerator
|
|
73: book
|
|
74: clock
|
|
75: vase
|
|
76: scissors
|
|
77: teddy bear
|
|
78: hair drier
|
|
79: toothbrush
|
|
|
|
# Download script/URL (optional)
|
|
download: |
|
|
from utils.general import download, Path
|
|
|
|
|
|
# Download labels
|
|
segments = False # segment or box labels
|
|
dir = Path(yaml['path']) # dataset root dir
|
|
url = 'https://github.com/ultralytics/yolov5/releases/download/v1.0/'
|
|
urls = [url + ('coco2017labels-segments.zip' if segments else 'coco2017labels.zip')] # labels
|
|
download(urls, dir=dir.parent)
|
|
|
|
# Download data
|
|
urls = ['http://images.cocodataset.org/zips/train2017.zip', # 19G, 118k images
|
|
'http://images.cocodataset.org/zips/val2017.zip', # 1G, 5k images
|
|
'http://images.cocodataset.org/zips/test2017.zip'] # 7G, 41k images (optional)
|
|
download(urls, dir=dir / 'images', threads=3)
|