train RTCDet-S

models/detectors/yolov7/README.md

@@ -3,7 +3,7 @@
 |    Model    |   Backbone    | Batch | Scale | AP<sup>val<br>0.5:0.95 | AP<sup>val<br>0.5 | FLOPs<br><sup>(G) | Params<br><sup>(M) | Weight |
 |-------------|---------------|-------|-------|------------------------|-------------------|-------------------|--------------------|--------|
 | YOLOv7-Tiny | ELANNet-Tiny  | 8xb16 |  640  |         39.5           |       58.5        |   22.6            |   7.9              | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov7_tiny_coco.pth) |
-| YOLOv7      | ELANNet-Large | 1xb16 |  640  |         48.0           |       67.5        |   144.6           |   44.0             | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov7_large_coco.pth) |
+| YOLOv7      | ELANNet-Large | 8xb16 |  640  |         49.5           |       68.8        |   144.6           |   44.0             | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov7_coco.pth) |
 | YOLOv7-X    | ELANNet-Huge  |       |  640  |                        |                   |                   |                    |  |
 
 - For training, we train `YOLOv7` and `YOLOv7-Tiny` with 300 epochs on 8 GPUs.

train_multi_gpus.sh

@@ -4,7 +4,7 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     -dist \
                                                     -d coco \
                                                     --root /data/datasets/ \
-                                                    -m yolov7 \
+                                                    -m rtcdet_s \
                                                     -bs 128 \
                                                     -size 640 \
                                                     --wp_epoch 3 \

train_single_gpu.sh

@@ -3,7 +3,7 @@ python train.py \
         --cuda \
         -d coco \
         --root /data/datasets/ \
-        -m yolov7_tiny \
+        -m rtcdet_s \
         -bs 8 \
         -size 640 \
         --wp_epoch 3 \