update

train_multi_gpus.sh

@@ -1,5 +1,4 @@
-# train YOLO with 4 GPUs
-# 使用 4 GPU来训练YOLO
+# -------------------------- Train YOLOX series --------------------------
 python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     --cuda \
                                                     -dist \
@@ -17,3 +16,22 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     --sybn \
                                                     --multi_scale \
                                                     #  --resume weights/coco/yolox_l/yolox_l_best.pth \
+
+# -------------------------- Train YOLOv1~v5 & v7 series --------------------------
+# python -m torch.distributed.run --nproc_per_node=8 train.py \
+#                                                     --cuda \
+#                                                     -dist \
+#                                                     -d coco \
+#                                                     --root /data/datasets/ \
+#                                                     -m yolov5_l\
+#                                                     -bs 128 \
+#                                                     -size 640 \
+#                                                     --wp_epoch 3 \
+#                                                     --max_epoch 300 \
+#                                                     --eval_epoch 10 \
+#                                                     --no_aug_epoch 10 \
+#                                                     --ema \
+#                                                     --fp16 \
+#                                                     --sybn \
+#                                                     --multi_scale \
+#                                                     #  --resume weights/coco/yolov5_l/yolov5_l_best.pth \

train_single_gpu.sh

@@ -1,4 +1,4 @@
-# Train YOLO
+# -------------------------- Train YOLOX series --------------------------
 python train.py \
         --cuda \
         -d coco \
@@ -9,11 +9,28 @@ python train.py \
         --wp_epoch 3 \
         --max_epoch 300 \
         --eval_epoch 10 \
-        --no_aug_epoch 20 \
+        --no_aug_epoch 15 \
         --grad_accumulate 8 \
         --ema \
         --fp16 \
         --multi_scale \
         # --resume weights/coco/yolox_m/yolox_m_best.pth \
-        # --pretrained weights/coco/yolo_free_medium/yolo_free_medium_39.46.pth \
         # --eval_first
+
+# -------------------------- Train YOLOv1~v5 & v7 series --------------------------
+# python train.py \
+#         --cuda \
+#         -d coco \
+#         --root /mnt/share/ssd2/dataset/ \
+#         -m yolov5_l \
+#         -bs 16 \
+#         -size 640 \
+#         --wp_epoch 3 \
+#         --max_epoch 300 \
+#         --eval_epoch 10 \
+#         --no_aug_epoch 10 \
+#         --ema \
+#         --fp16 \
+#         --multi_scale \
+#         # --resume weights/coco/yolov5_l/yolov5_l_best.pth \
+#         # --eval_first