Pārlūkot izejas kodu

release new ckpt of YOLOv7-Tiny

yjh0410 2 gadi atpakaļ
vecāks
revīzija
2ffd0befc2
2 mainītis faili ar 3 papildinājumiem un 3 dzēšanām
  1. 2 2
      models/detectors/yolov7/README.md
  2. 1 1
      train_multi_gpus.sh

+ 2 - 2
models/detectors/yolov7/README.md

@@ -2,8 +2,8 @@
 
 |    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  | 1xb16 |  640  |         38.0           |       56.8        |   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-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 | 8xb16 |  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-X    | ELANNet-Huge  | 8xb8  |  640  |                        |                   |                   |                    |  |
 
 - For training, we train YOLOv7 and YOLOv7-Tiny with 300 epochs on COCO.

+ 1 - 1
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_tiny\
+                                                    -m yolov7 \
                                                     -bs 128 \
                                                     -size 640 \
                                                     --wp_epoch 3 \