train YOLOv8-N

config/model_config/rtcdet_config.py

@@ -62,7 +62,7 @@ rtcdet_cfg = {
         # ---------------- Model config ----------------
         ## Backbone
         'backbone': 'elannet',
-        'pretrained': False,
+        'pretrained': True,
         'bk_act': 'silu',
         'bk_norm': 'BN',
         'bk_depthwise': False,

config/model_config/yolov8_config.py

@@ -51,7 +51,7 @@ yolov8_cfg = {
         'loss_dfl_weight': 1.5,
         'loss_box_aux': False,
         # ---------------- Train config ----------------
-        'trainer_type': 'rtcdet',
+        'trainer_type': 'yolov8',
     },
 
     'yolov8_l':{

train_multi_gpus.sh

@@ -1,60 +1,60 @@
 # -------------------------- Train RTCDet --------------------------
-python -m torch.distributed.run --nproc_per_node=8 train.py \
-                                                    --cuda \
-                                                    -dist \
-                                                    -d coco \
-                                                    --root /data/datasets/ \
-                                                    -m rtcdet_n \
-                                                    -bs 64 \
-                                                    -size 640 \
-                                                    --wp_epoch 3 \
-                                                    --max_epoch 300 \
-                                                    --eval_epoch 10 \
-                                                    --no_aug_epoch 20 \
-                                                    --ema \
-                                                    --fp16 \
-                                                    --sybn \
-                                                    --multi_scale \
-                                                    --save_folder weights/ \
-                                                    # --load_cache \
-                                                    # --resume weights/coco/yolox_l/yolox_l_best.pth \
-
-# -------------------------- Train YOLOX & YOLOv7 --------------------------
 # python -m torch.distributed.run --nproc_per_node=8 train.py \
 #                                                     --cuda \
 #                                                     -dist \
 #                                                     -d coco \
 #                                                     --root /data/datasets/ \
-#                                                     -m rtcdet_n \
-#                                                     -bs 64 \
+#                                                     -m rtcdet_s \
+#                                                     -bs 128 \
 #                                                     -size 640 \
 #                                                     --wp_epoch 3 \
 #                                                     --max_epoch 300 \
 #                                                     --eval_epoch 10 \
-#                                                     --no_aug_epoch 15 \
+#                                                     --no_aug_epoch 20 \
 #                                                     --ema \
 #                                                     --fp16 \
 #                                                     --sybn \
 #                                                     --multi_scale \
+#                                                     --save_folder weights/ \
 #                                                     # --load_cache \
 #                                                     # --resume weights/coco/yolox_l/yolox_l_best.pth \
 
-# # -------------------------- Train YOLOv1~v5 --------------------------
+# -------------------------- Train YOLOX & YOLOv7 --------------------------
 # python -m torch.distributed.run --nproc_per_node=8 train.py \
 #                                                     --cuda \
 #                                                     -dist \
 #                                                     -d coco \
 #                                                     --root /data/datasets/ \
-#                                                     -m yolov8_n\
-#                                                     -bs 256 \
+#                                                     -m rtcdet_n \
+#                                                     -bs 64 \
 #                                                     -size 640 \
 #                                                     --wp_epoch 3 \
-#                                                     --max_epoch 500 \
+#                                                     --max_epoch 300 \
 #                                                     --eval_epoch 10 \
-#                                                     --no_aug_epoch 10 \
+#                                                     --no_aug_epoch 15 \
 #                                                     --ema \
 #                                                     --fp16 \
 #                                                     --sybn \
 #                                                     --multi_scale \
-#                                                     # --load_cache
-#                                                     # --resume weights/coco/yolov5_l/yolov5_l_best.pth \
+#                                                     # --load_cache \
+#                                                     # --resume weights/coco/yolox_l/yolox_l_best.pth \
+
+# -------------------------- Train YOLOv1~v5 --------------------------
+python -m torch.distributed.run --nproc_per_node=8 train.py \
+                                                    --cuda \
+                                                    -dist \
+                                                    -d coco \
+                                                    --root /data/datasets/ \
+                                                    -m yolov8_n\
+                                                    -bs 256 \
+                                                    -size 640 \
+                                                    --wp_epoch 3 \
+                                                    --max_epoch 500 \
+                                                    --eval_epoch 10 \
+                                                    --no_aug_epoch 10 \
+                                                    --ema \
+                                                    --fp16 \
+                                                    --sybn \
+                                                    --multi_scale \
+                                                    # --load_cache
+                                                    # --resume weights/coco/yolov5_l/yolov5_l_best.pth \