2个经典模型复现-Alexnet 和 yolov5

程治玮/yolov5-5.0/.gitignore

@@ -1,252 +1,252 @@
-# Repo-specific GitIgnore ----------------------------------------------------------------------------------------------
+# # Repo-specific GitIgnore ----------------------------------------------------------------------------------------------
-*.jpg
+# *.jpg
-*.jpeg
+# *.jpeg
-*.png
+# *.png
-*.bmp
+# *.bmp
-*.tif
+# *.tif
-*.tiff
+# *.tiff
-*.heic
+# *.heic
-*.JPG
+# *.JPG
-*.JPEG
+# *.JPEG
-*.PNG
+# *.PNG
-*.BMP
+# *.BMP
-*.TIF
+# *.TIF
-*.TIFF
+# *.TIFF
-*.HEIC
+# *.HEIC
-*.mp4
+# *.mp4
-*.mov
+# *.mov
-*.MOV
+# *.MOV
-*.avi
+# *.avi
-*.data
+# *.data
-*.json
+# *.json
 
-*.cfg
+# *.cfg
-!cfg/yolov3*.cfg
+# !cfg/yolov3*.cfg
 
-storage.googleapis.com
+# storage.googleapis.com
-runs/*
+# runs/*
-data/*
+# data/*
-!data/images/zidane.jpg
+# !data/images/zidane.jpg
-!data/images/bus.jpg
+# !data/images/bus.jpg
-!data/coco.names
+# !data/coco.names
-!data/coco_paper.names
+# !data/coco_paper.names
-!data/coco.data
+# !data/coco.data
-!data/coco_*.data
+# !data/coco_*.data
-!data/coco_*.txt
+# !data/coco_*.txt
-!data/trainvalno5k.shapes
+# !data/trainvalno5k.shapes
-!data/*.sh
+# !data/*.sh
 
-pycocotools/*
+# pycocotools/*
-results*.txt
+# results*.txt
-gcp_test*.sh
+# gcp_test*.sh
 
-# Datasets -------------------------------------------------------------------------------------------------------------
+# # Datasets -------------------------------------------------------------------------------------------------------------
-coco/
+# coco/
-coco128/
+# coco128/
-VOC/
+# VOC/
 
-# MATLAB GitIgnore -----------------------------------------------------------------------------------------------------
+# # MATLAB GitIgnore -----------------------------------------------------------------------------------------------------
-*.m~
+# *.m~
-*.mat
+# *.mat
-!targets*.mat
+# !targets*.mat
 
-# Neural Network weights -----------------------------------------------------------------------------------------------
+# # Neural Network weights -----------------------------------------------------------------------------------------------
-*.weights
+# *.weights
-*.pt
+# *.pt
-*.onnx
+# *.onnx
-*.mlmodel
+# *.mlmodel
-*.torchscript
+# *.torchscript
-darknet53.conv.74
+# darknet53.conv.74
-yolov3-tiny.conv.15
+# yolov3-tiny.conv.15
 
-# GitHub Python GitIgnore ----------------------------------------------------------------------------------------------
+# # GitHub Python GitIgnore ----------------------------------------------------------------------------------------------
-# Byte-compiled / optimized / DLL files
+# # Byte-compiled / optimized / DLL files
-__pycache__/
+# __pycache__/
-*.py[cod]
+# *.py[cod]
-*$py.class
+# *$py.class
 
-# C extensions
+# # C extensions
-*.so
+# *.so
 
-# Distribution / packaging
+# # Distribution / packaging
-.Python
+# .Python
-env/
+# env/
-build/
+# build/
-develop-eggs/
+# develop-eggs/
-dist/
+# dist/
-downloads/
+# downloads/
-eggs/
+# eggs/
-.eggs/
+# .eggs/
-lib/
+# lib/
-lib64/
+# lib64/
-parts/
+# parts/
-sdist/
+# sdist/
-var/
+# var/
-wheels/
+# wheels/
-*.egg-info/
+# *.egg-info/
-wandb/
+# wandb/
-.installed.cfg
+# .installed.cfg
-*.egg
+# *.egg
 
 
-# PyInstaller
+# # PyInstaller
-#  Usually these files are written by a python script from a template
+# #  Usually these files are written by a python script from a template
-#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+# #  before PyInstaller builds the exe, so as to inject date/other infos into it.
-*.manifest
+# *.manifest
-*.spec
+# *.spec
 
-# Installer logs
+# # Installer logs
-pip-log.txt
+# pip-log.txt
-pip-delete-this-directory.txt
+# pip-delete-this-directory.txt
 
-# Unit test / coverage reports
+# # Unit test / coverage reports
-htmlcov/
+# htmlcov/
-.tox/
+# .tox/
-.coverage
+# .coverage
-.coverage.*
+# .coverage.*
-.cache
+# .cache
-nosetests.xml
+# nosetests.xml
-coverage.xml
+# coverage.xml
-*.cover
+# *.cover
-.hypothesis/
+# .hypothesis/
 
-# Translations
+# # Translations
-*.mo
+# *.mo
-*.pot
+# *.pot
 
-# Django stuff:
+# # Django stuff:
-*.log
+# *.log
-local_settings.py
+# local_settings.py
 
-# Flask stuff:
+# # Flask stuff:
-instance/
+# instance/
-.webassets-cache
+# .webassets-cache
 
-# Scrapy stuff:
+# # Scrapy stuff:
-.scrapy
+# .scrapy
 
-# Sphinx documentation
+# # Sphinx documentation
-docs/_build/
+# docs/_build/
 
-# PyBuilder
+# # PyBuilder
-target/
+# target/
 
-# Jupyter Notebook
+# # Jupyter Notebook
-.ipynb_checkpoints
+# .ipynb_checkpoints
 
-# pyenv
+# # pyenv
-.python-version
+# .python-version
 
-# celery beat schedule file
+# # celery beat schedule file
-celerybeat-schedule
+# celerybeat-schedule
 
-# SageMath parsed files
+# # SageMath parsed files
-*.sage.py
+# *.sage.py
 
-# dotenv
+# # dotenv
-.env
+# .env
 
-# virtualenv
+# # virtualenv
-.venv*
+# .venv*
-venv*/
+# venv*/
-ENV*/
+# ENV*/
 
-# Spyder project settings
+# # Spyder project settings
-.spyderproject
+# .spyderproject
-.spyproject
+# .spyproject
 
-# Rope project settings
+# # Rope project settings
-.ropeproject
+# .ropeproject
 
-# mkdocs documentation
+# # mkdocs documentation
-/site
+# /site
 
-# mypy
+# # mypy
-.mypy_cache/
+# .mypy_cache/
 
 
-# https://github.com/github/gitignore/blob/master/Global/macOS.gitignore -----------------------------------------------
+# # https://github.com/github/gitignore/blob/master/Global/macOS.gitignore -----------------------------------------------
 
-# General
+# # General
-.DS_Store
+# .DS_Store
-.AppleDouble
+# .AppleDouble
-.LSOverride
+# .LSOverride
 
-# Icon must end with two \r
+# # Icon must end with two \r
-Icon
+# Icon
-Icon?
+# Icon?
 
-# Thumbnails
+# # Thumbnails
-._*
+# ._*
 
-# Files that might appear in the root of a volume
+# # Files that might appear in the root of a volume
-.DocumentRevisions-V100
+# .DocumentRevisions-V100
-.fseventsd
+# .fseventsd
-.Spotlight-V100
+# .Spotlight-V100
-.TemporaryItems
+# .TemporaryItems
-.Trashes
+# .Trashes
-.VolumeIcon.icns
+# .VolumeIcon.icns
-.com.apple.timemachine.donotpresent
+# .com.apple.timemachine.donotpresent
 
-# Directories potentially created on remote AFP share
+# # Directories potentially created on remote AFP share
-.AppleDB
+# .AppleDB
-.AppleDesktop
+# .AppleDesktop
-Network Trash Folder
+# Network Trash Folder
-Temporary Items
+# Temporary Items
-.apdisk
+# .apdisk
 
 
-# https://github.com/github/gitignore/blob/master/Global/JetBrains.gitignore
+# # https://github.com/github/gitignore/blob/master/Global/JetBrains.gitignore
-# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
+# # Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
-# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+# # Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
 
-# User-specific stuff:
+# # User-specific stuff:
-.idea/*
+# .idea/*
-.idea/**/workspace.xml
+# .idea/**/workspace.xml
-.idea/**/tasks.xml
+# .idea/**/tasks.xml
-.idea/dictionaries
+# .idea/dictionaries
-.html  # Bokeh Plots
+# .html  # Bokeh Plots
-.pg  # TensorFlow Frozen Graphs
+# .pg  # TensorFlow Frozen Graphs
-.avi # videos
+# .avi # videos
 
-# Sensitive or high-churn files:
+# # Sensitive or high-churn files:
-.idea/**/dataSources/
+# .idea/**/dataSources/
-.idea/**/dataSources.ids
+# .idea/**/dataSources.ids
-.idea/**/dataSources.local.xml
+# .idea/**/dataSources.local.xml
-.idea/**/sqlDataSources.xml
+# .idea/**/sqlDataSources.xml
-.idea/**/dynamic.xml
+# .idea/**/dynamic.xml
-.idea/**/uiDesigner.xml
+# .idea/**/uiDesigner.xml
 
-# Gradle:
+# # Gradle:
-.idea/**/gradle.xml
+# .idea/**/gradle.xml
-.idea/**/libraries
+# .idea/**/libraries
 
-# CMake
+# # CMake
-cmake-build-debug/
+# cmake-build-debug/
-cmake-build-release/
+# cmake-build-release/
 
-# Mongo Explorer plugin:
+# # Mongo Explorer plugin:
-.idea/**/mongoSettings.xml
+# .idea/**/mongoSettings.xml
 
-## File-based project format:
+# ## File-based project format:
-*.iws
+# *.iws
 
-## Plugin-specific files:
+# ## Plugin-specific files:
 
-# IntelliJ
+# # IntelliJ
-out/
+# out/
 
-# mpeltonen/sbt-idea plugin
+# # mpeltonen/sbt-idea plugin
-.idea_modules/
+# .idea_modules/
 
-# JIRA plugin
+# # JIRA plugin
-atlassian-ide-plugin.xml
+# atlassian-ide-plugin.xml
 
-# Cursive Clojure plugin
+# # Cursive Clojure plugin
-.idea/replstate.xml
+# .idea/replstate.xml
 
-# Crashlytics plugin (for Android Studio and IntelliJ)
+# # Crashlytics plugin (for Android Studio and IntelliJ)
-com_crashlytics_export_strings.xml
+# com_crashlytics_export_strings.xml
-crashlytics.properties
+# crashlytics.properties
-crashlytics-build.properties
+# crashlytics-build.properties
-fabric.properties
+# fabric.properties

程治玮/yolov5-5.0/data/argoverse_hd.yaml

@@ -0,0 +1,21 @@
+# Argoverse-HD dataset (ring-front-center camera) http://www.cs.cmu.edu/~mengtial/proj/streaming/
+# Train command: python train.py --data argoverse_hd.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /argoverse
+#     /yolov5
+
+
+# download command/URL (optional)
+download: bash data/scripts/get_argoverse_hd.sh
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+train: ../argoverse/Argoverse-1.1/images/train/  # 39384 images
+val: ../argoverse/Argoverse-1.1/images/val/  # 15062 iamges
+test: ../argoverse/Argoverse-1.1/images/test/  # Submit to: https://eval.ai/web/challenges/challenge-page/800/overview
+
+# number of classes
+nc: 8
+
+# class names
+names: [ 'person',  'bicycle',  'car',  'motorcycle',  'bus',  'truck',  'traffic_light',  'stop_sign' ]

程治玮/yolov5-5.0/data/coco.yaml

@@ -0,0 +1,35 @@
+# COCO 2017 dataset http://cocodataset.org
+# Train command: python train.py --data coco.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /coco
+#     /yolov5
+
+
+# download command/URL (optional)
+download: bash data/scripts/get_coco.sh
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+train: ../coco/train2017.txt  # 118287 images
+val: ../coco/val2017.txt  # 5000 images
+test: ../coco/test-dev2017.txt  # 20288 of 40670 images, submit to https://competitions.codalab.org/competitions/20794
+
+# number of classes
+nc: 80
+
+# class names
+names: [ 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
+         'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
+         'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
+         'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
+         'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
+         'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
+         'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
+         'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
+         'hair drier', 'toothbrush' ]
+
+# Print classes
+# with open('data/coco.yaml') as f:
+#   d = yaml.load(f, Loader=yaml.FullLoader)  # dict
+#   for i, x in enumerate(d['names']):
+#     print(i, x)

程治玮/yolov5-5.0/data/coco128.yaml

@@ -0,0 +1,30 @@
+# COCO 2017 dataset http://cocodataset.org - first 128 training images
+# Train command: python train.py --data coco128.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /coco128
+#     /yolov5
+
+#
+## download command/URL (optional)
+download: https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+train: ../coco128/images/train2017/  # 128 images
+val: ../coco128/images/train2017/  # 128 images
+
+
+# number of classes
+nc: 80
+
+
+# class names
+names: [ 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
+         'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
+         'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
+         'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
+         'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
+         'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
+         'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
+         'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
+         'hair drier', 'toothbrush' ]

程治玮/yolov5-5.0/data/hyp.finetune.yaml

@@ -0,0 +1,38 @@
+# Hyperparameters for VOC finetuning
+# python train.py --batch 64 --weights yolov5m.pt --data voc.yaml --img 512 --epochs 50
+# See tutorials for hyperparameter evolution https://github.com/ultralytics/yolov5#tutorials
+
+
+# Hyperparameter Evolution Results
+# Generations: 306
+#                   P         R     mAP.5 mAP.5:.95       box       obj       cls
+# Metrics:        0.6     0.936     0.896     0.684    0.0115   0.00805   0.00146
+
+lr0: 0.0032
+lrf: 0.12
+momentum: 0.843
+weight_decay: 0.00036
+warmup_epochs: 2.0
+warmup_momentum: 0.5
+warmup_bias_lr: 0.05
+box: 0.0296
+cls: 0.243
+cls_pw: 0.631
+obj: 0.301
+obj_pw: 0.911
+iou_t: 0.2
+anchor_t: 2.91
+# anchors: 3.63
+fl_gamma: 0.0
+hsv_h: 0.0138
+hsv_s: 0.664
+hsv_v: 0.464
+degrees: 0.373
+translate: 0.245
+scale: 0.898
+shear: 0.602
+perspective: 0.0
+flipud: 0.00856
+fliplr: 0.5
+mosaic: 1.0
+mixup: 0.243

程治玮/yolov5-5.0/data/hyp.scratch.yaml

@@ -0,0 +1,33 @@
+# Hyperparameters for COCO training from scratch
+# python train.py --batch 40 --cfg yolov5m.yaml --weights '' --data coco.yaml --img 640 --epochs 300
+# See tutorials for hyperparameter evolution https://github.com/ultralytics/yolov5#tutorials
+
+
+lr0: 0.01  # initial learning rate (SGD=1E-2, Adam=1E-3)
+lrf: 0.2  # final OneCycleLR learning rate (lr0 * lrf)
+momentum: 0.937  # SGD momentum/Adam beta1
+weight_decay: 0.0005  # optimizer weight decay 5e-4
+warmup_epochs: 3.0  # warmup epochs (fractions ok)
+warmup_momentum: 0.8  # warmup initial momentum
+warmup_bias_lr: 0.1  # warmup initial bias lr
+box: 0.05  # box loss gain
+cls: 0.5  # cls loss gain
+cls_pw: 1.0  # cls BCELoss positive_weight
+obj: 1.0  # obj loss gain (scale with pixels)
+obj_pw: 1.0  # obj BCELoss positive_weight
+iou_t: 0.20  # IoU training threshold
+anchor_t: 4.0  # anchor-multiple threshold
+# anchors: 3  # anchors per output layer (0 to ignore)
+fl_gamma: 0.0  # focal loss gamma (efficientDet default gamma=1.5)
+hsv_h: 0.015  # image HSV-Hue augmentation (fraction)
+hsv_s: 0.7  # image HSV-Saturation augmentation (fraction)
+hsv_v: 0.4  # image HSV-Value augmentation (fraction)
+degrees: 0.0  # image rotation (+/- deg)
+translate: 0.1  # image translation (+/- fraction)
+scale: 0.5  # image scale (+/- gain)
+shear: 0.0  # image shear (+/- deg)
+perspective: 0.0  # image perspective (+/- fraction), range 0-0.001
+flipud: 0.0  # image flip up-down (probability)
+fliplr: 0.5  # image flip left-right (probability)
+mosaic: 1.0  # image mosaic (probability)
+mixup: 0.0  # image mixup (probability)

程治玮/yolov5-5.0/data/mydata.yaml

@@ -0,0 +1,32 @@
+# COCO 2017 dataset http://cocodataset.org - first 128 training images
+# Train command: python train.py --data coco128.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /coco128
+#     /yolov5
+
+#
+## download command/URL (optional)
+#download: https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+#train: ../coco128/images/train2017/  # 128 images
+#val: ../coco128/images/train2017/  # 128 images
+train: mydata\images\train
+val: mydata\images\train
+
+# number of classes
+#nc: 80
+nc: 1
+## class names
+#names: [ 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
+#         'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
+#         'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
+#         'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
+#         'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
+#         'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
+#         'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
+#         'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear',
+#         'hair drier', 'toothbrush' ]
+
+names: ['wind']

程治玮/yolov5-5.0/data/scripts/get_argoverse_hd.sh

@@ -0,0 +1,62 @@
+#!/bin/bash
+# Argoverse-HD dataset (ring-front-center camera) http://www.cs.cmu.edu/~mengtial/proj/streaming/
+# Download command: bash data/scripts/get_argoverse_hd.sh
+# Train command: python train.py --data argoverse_hd.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /argoverse
+#     /yolov5
+
+# Download/unzip images
+d='../argoverse/' # unzip directory
+mkdir $d
+url=https://argoverse-hd.s3.us-east-2.amazonaws.com/
+f=Argoverse-HD-Full.zip
+curl -L $url$f -o $f && unzip -q $f -d $d && rm $f &# download, unzip, remove in background
+wait                                              # finish background tasks
+
+cd ../argoverse/Argoverse-1.1/
+ln -s tracking images
+
+cd ../Argoverse-HD/annotations/
+
+python3 - "$@" <<END
+import json
+from pathlib import Path
+
+annotation_files = ["train.json", "val.json"]
+print("Converting annotations to YOLOv5 format...")
+
+for val in annotation_files:
+    a = json.load(open(val, "rb"))
+
+    label_dict = {}
+    for annot in a['annotations']:
+        img_id = annot['image_id']
+        img_name = a['images'][img_id]['name']
+        img_label_name = img_name[:-3] + "txt"
+
+        obj_class = annot['category_id']
+        x_center, y_center, width, height = annot['bbox']
+        x_center = (x_center + width / 2) / 1920.  # offset and scale
+        y_center = (y_center + height / 2) / 1200.  # offset and scale
+        width /= 1920.  # scale
+        height /= 1200.  # scale
+
+        img_dir = "./labels/" + a['seq_dirs'][a['images'][annot['image_id']]['sid']]
+
+        Path(img_dir).mkdir(parents=True, exist_ok=True)
+
+        if img_dir + "/" + img_label_name not in label_dict:
+            label_dict[img_dir + "/" + img_label_name] = []
+
+        label_dict[img_dir + "/" + img_label_name].append(f"{obj_class} {x_center} {y_center} {width} {height}\n")
+
+    for filename in label_dict:
+        with open(filename, "w") as file:
+            for string in label_dict[filename]:
+                file.write(string)
+
+END
+
+mv ./labels ../../Argoverse-1.1/

程治玮/yolov5-5.0/data/scripts/get_coco.sh

@@ -0,0 +1,27 @@
+#!/bin/bash
+# COCO 2017 dataset http://cocodataset.org
+# Download command: bash data/scripts/get_coco.sh
+# Train command: python train.py --data coco.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /coco
+#     /yolov5
+
+# Download/unzip labels
+d='../' # unzip directory
+url=https://github.com/ultralytics/yolov5/releases/download/v1.0/
+f='coco2017labels.zip' # or 'coco2017labels-segments.zip', 68 MB
+echo 'Downloading' $url$f ' ...'
+curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
+
+# Download/unzip images
+d='../coco/images' # unzip directory
+url=http://images.cocodataset.org/zips/
+f1='train2017.zip' # 19G, 118k images
+f2='val2017.zip'   # 1G, 5k images
+f3='test2017.zip'  # 7G, 41k images (optional)
+for f in $f1 $f2; do
+  echo 'Downloading' $url$f '...'
+  curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
+done
+wait # finish background tasks

程治玮/yolov5-5.0/data/scripts/get_voc.sh

@@ -0,0 +1,139 @@
+#!/bin/bash
+# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
+# Download command: bash data/scripts/get_voc.sh
+# Train command: python train.py --data voc.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /VOC
+#     /yolov5
+
+start=$(date +%s)
+mkdir -p ../tmp
+cd ../tmp/
+
+# Download/unzip images and labels
+d='.' # unzip directory
+url=https://github.com/ultralytics/yolov5/releases/download/v1.0/
+f1=VOCtrainval_06-Nov-2007.zip # 446MB, 5012 images
+f2=VOCtest_06-Nov-2007.zip     # 438MB, 4953 images
+f3=VOCtrainval_11-May-2012.zip # 1.95GB, 17126 images
+for f in $f3 $f2 $f1; do
+  echo 'Downloading' $url$f '...' 
+  curl -L $url$f -o $f && unzip -q $f -d $d && rm $f & # download, unzip, remove in background
+done
+wait # finish background tasks
+
+end=$(date +%s)
+runtime=$((end - start))
+echo "Completed in" $runtime "seconds"
+
+echo "Splitting dataset..."
+python3 - "$@" <<END
+import xml.etree.ElementTree as ET
+import pickle
+import os
+from os import listdir, getcwd
+from os.path import join
+
+sets=[('2012', 'train'), ('2012', 'val'), ('2007', 'train'), ('2007', 'val'), ('2007', 'test')]
+
+classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]
+
+
+def convert(size, box):
+    dw = 1./(size[0])
+    dh = 1./(size[1])
+    x = (box[0] + box[1])/2.0 - 1
+    y = (box[2] + box[3])/2.0 - 1
+    w = box[1] - box[0]
+    h = box[3] - box[2]
+    x = x*dw
+    w = w*dw
+    y = y*dh
+    h = h*dh
+    return (x,y,w,h)
+
+def convert_annotation(year, image_id):
+    in_file = open('VOCdevkit/VOC%s/Annotations/%s.xml'%(year, image_id))
+    out_file = open('VOCdevkit/VOC%s/labels/%s.txt'%(year, image_id), 'w')
+    tree=ET.parse(in_file)
+    root = tree.getroot()
+    size = root.find('size')
+    w = int(size.find('width').text)
+    h = int(size.find('height').text)
+
+    for obj in root.iter('object'):
+        difficult = obj.find('difficult').text
+        cls = obj.find('name').text
+        if cls not in classes or int(difficult)==1:
+            continue
+        cls_id = classes.index(cls)
+        xmlbox = obj.find('bndbox')
+        b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text), float(xmlbox.find('ymax').text))
+        bb = convert((w,h), b)
+        out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')
+
+wd = getcwd()
+
+for year, image_set in sets:
+    if not os.path.exists('VOCdevkit/VOC%s/labels/'%(year)):
+        os.makedirs('VOCdevkit/VOC%s/labels/'%(year))
+    image_ids = open('VOCdevkit/VOC%s/ImageSets/Main/%s.txt'%(year, image_set)).read().strip().split()
+    list_file = open('%s_%s.txt'%(year, image_set), 'w')
+    for image_id in image_ids:
+        list_file.write('%s/VOCdevkit/VOC%s/JPEGImages/%s.jpg\n'%(wd, year, image_id))
+        convert_annotation(year, image_id)
+    list_file.close()
+
+END
+
+cat 2007_train.txt 2007_val.txt 2012_train.txt 2012_val.txt >train.txt
+cat 2007_train.txt 2007_val.txt 2007_test.txt 2012_train.txt 2012_val.txt >train.all.txt
+
+python3 - "$@" <<END
+
+import shutil
+import os
+os.system('mkdir ../VOC/')
+os.system('mkdir ../VOC/images')
+os.system('mkdir ../VOC/images/train')
+os.system('mkdir ../VOC/images/val')
+
+os.system('mkdir ../VOC/labels')
+os.system('mkdir ../VOC/labels/train')
+os.system('mkdir ../VOC/labels/val')
+
+import os
+print(os.path.exists('../tmp/train.txt'))
+f = open('../tmp/train.txt', 'r')
+lines = f.readlines()
+
+for line in lines:
+    line = "/".join(line.split('/')[-5:]).strip()
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/images/train")
+        
+    line = line.replace('JPEGImages', 'labels')
+    line = line.replace('jpg', 'txt')
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/labels/train")
+
+
+print(os.path.exists('../tmp/2007_test.txt'))
+f = open('../tmp/2007_test.txt', 'r')
+lines = f.readlines()
+
+for line in lines:
+    line = "/".join(line.split('/')[-5:]).strip()
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/images/val")
+        
+    line = line.replace('JPEGImages', 'labels')
+    line = line.replace('jpg', 'txt')
+    if (os.path.exists("../" + line)):
+        os.system("cp ../"+ line + " ../VOC/labels/val")
+
+END
+
+rm -rf ../tmp # remove temporary directory
+echo "VOC download done."

程治玮/yolov5-5.0/data/voc.yaml

@@ -0,0 +1,21 @@
+# PASCAL VOC dataset http://host.robots.ox.ac.uk/pascal/VOC/
+# Train command: python train.py --data voc.yaml
+# Default dataset location is next to /yolov5:
+#   /parent_folder
+#     /VOC
+#     /yolov5
+
+
+# download command/URL (optional)
+download: bash data/scripts/get_voc.sh
+
+# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
+train: ../VOC/images/train/  # 16551 images
+val: ../VOC/images/val/  # 4952 images
+
+# number of classes
+nc: 20
+
+# class names
+names: [ 'aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog',
+         'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor' ]