Ver Fonte

train YOLOvx-L on COCO

yjh0410 há 2 anos atrás
pai
commit
6619af6ef2
2 ficheiros alterados com 4 adições e 4 exclusões
  1. 2 2
      README.md
  2. 2 2
      train_ddp.sh

+ 2 - 2
README.md

@@ -137,8 +137,8 @@ python train.py --cuda -d coco --root path/to/COCO -m yolov1 -bs 16 --max_epoch
 |---------|---------------|-------|-------|------------------------|-------------------|-------------------|--------------------|--------|
 | YOLOX-N | CSPDarkNet-N  |  640  |  300  |         31.1           |       49.5        |   7.5             |   2.3              | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolox_n_coco.pth) |
 | YOLOX-S | CSPDarkNet-S  |  640  |  300  |         39.0           |       58.8        |   26.8            |   8.9              | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolox_s_coco.pth) |
-| YOLOX-M | CSPDarkNet-M  |  640  |  300  |                        |                   |   74.3            |   25.4             |  |
-| YOLOX-L | CSPDarkNet-L  |  640  |  300  |         46.9           |       65.9        |   155.4           |   54.2             |  |
+| YOLOX-M | CSPDarkNet-M  |  640  |  300  |         44.6           |       63.8        |   74.3            |   25.4             | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolox_m_coco.pth) |
+| YOLOX-L | CSPDarkNet-L  |  640  |  300  |         46.9           |       65.9        |   155.4           |   54.2             | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolox_l_coco.pth) |
 
 *For **YOLOX-M** and **YOLOX-L**, increasing the batch size may improve performance. Due to my computing resources, I can only set the batch size to 16.*
 

+ 2 - 2
train_ddp.sh

@@ -1,11 +1,11 @@
 # train YOLO with 8 GPUs
-# 使用4张GPU来训练YOLO
+# 使用8张GPU来训练YOLO
 python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     --cuda \
                                                     -dist \
                                                     -d coco \
                                                     --root /data/datasets/ \
-                                                    -m yolox_l \
+                                                    -m yolovx_l \
                                                     -bs 128 \
                                                     -size 640 \
                                                     --wp_epoch 3 \