完成训练模块的转移

yolov5/data/scripts/get_imagenet1000.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+# YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
+# Download ILSVRC2012 ImageNet dataset https://image-net.org
+# Example usage: bash data/scripts/get_imagenet.sh
+# parent
+# ├── yolov5
+# └── datasets
+#     └── imagenet  ← downloads here
+
+# Arguments (optional) Usage: bash data/scripts/get_imagenet.sh --train --val
+if [ "$#" -gt 0 ]; then
+  for opt in "$@"; do
+    case "${opt}" in
+    --train) train=true ;;
+    --val) val=true ;;
+    esac
+  done
+else
+  train=true
+  val=true
+fi
+
+# Make dir
+d='../datasets/imagenet1000' # unzip directory
+mkdir -p $d && cd $d
+
+# Download/unzip train
+wget https://github.com/ultralytics/yolov5/releases/download/v1.0/imagenet1000.zip
+unzip imagenet1000.zip && rm imagenet1000.zip