mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
33712d6dd0
* Create GlobalWheat2020.yaml * Update and rename visdrone.yaml to VisDrone.yaml * Update GlobalWheat2020.yaml
56 lines
2.0 KiB
YAML
56 lines
2.0 KiB
YAML
# Global Wheat 2020 dataset http://www.global-wheat.com/
|
|
# Train command: python train.py --data GlobalWheat2020.yaml
|
|
# Default dataset location is next to YOLOv5:
|
|
# /parent_folder
|
|
# /datasets/GlobalWheat2020
|
|
# /yolov5
|
|
|
|
|
|
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
|
|
train: # 3422 images
|
|
- ../datasets/GlobalWheat2020/images/arvalis_1
|
|
- ../datasets/GlobalWheat2020/images/arvalis_2
|
|
- ../datasets/GlobalWheat2020/images/arvalis_3
|
|
- ../datasets/GlobalWheat2020/images/ethz_1
|
|
- ../datasets/GlobalWheat2020/images/rres_1
|
|
- ../datasets/GlobalWheat2020/images/inrae_1
|
|
- ../datasets/GlobalWheat2020/images/usask_1
|
|
|
|
val: # 748 images (WARNING: train set contains ethz_1)
|
|
- ../datasets/GlobalWheat2020/images/ethz_1
|
|
|
|
test: # 1276
|
|
- ../datasets/GlobalWheat2020/images/utokyo_1
|
|
- ../datasets/GlobalWheat2020/images/utokyo_2
|
|
- ../datasets/GlobalWheat2020/images/nau_1
|
|
- ../datasets/GlobalWheat2020/images/uq_1
|
|
|
|
# number of classes
|
|
nc: 1
|
|
|
|
# class names
|
|
names: [ 'wheat_head' ]
|
|
|
|
|
|
# download command/URL (optional) --------------------------------------------------------------------------------------
|
|
download: |
|
|
from utils.general import download, Path
|
|
|
|
# Download
|
|
dir = Path('../datasets/GlobalWheat2020') # dataset directory
|
|
urls = ['https://zenodo.org/record/4298502/files/global-wheat-codalab-official.zip',
|
|
'https://github.com/ultralytics/yolov5/releases/download/v1.0/GlobalWheat2020_labels.zip']
|
|
download(urls, dir=dir)
|
|
|
|
# Make Directories
|
|
for p in 'annotations', 'images', 'labels':
|
|
(dir / p).mkdir(parents=True, exist_ok=True)
|
|
|
|
# Move
|
|
for p in 'arvalis_1', 'arvalis_2', 'arvalis_3', 'ethz_1', 'rres_1', 'inrae_1', 'usask_1', \
|
|
'utokyo_1', 'utokyo_2', 'nau_1', 'uq_1':
|
|
(dir / p).rename(dir / 'images' / p) # move to /images
|
|
f = (dir / p).with_suffix('.json') # json file
|
|
if f.exists():
|
|
f.rename((dir / 'annotations' / p).with_suffix('.json')) # move to /annotations
|