|
|
@@ -72,7 +72,7 @@ python train.py --cuda -d voc --root path/to/VOCdevkit -v yolov1 -bs 16 --max_ep
|
|
|
| YOLOv4 | CSPDarkNet-53 | 640 | √ | 150 | 83.6 | | 162.7 | 61.5 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov4_voc.pth) |
|
|
|
| YOLOv5 | CSPDarkNet-L | 640 | √ | 150 | | | | | |
|
|
|
| YOLOX | CSPDarkNet-L | 640 | √ | 150 | 84.6 | | 155.4 | 54.2 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolox_voc.pth) |
|
|
|
-| YOLOv7 | ELANNet | 640 | √ | 150 | 85.5 | | 144.6 | 44.0 | |
|
|
|
+| YOLOv7 | ELANNet | 640 | √ | 150 | 85.5 | | 144.6 | 44.0 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov7_voc.pth) |
|
|
|
|
|
|
*All models are trained with ImageNet pretrained weight (IP). All FLOPs are measured with a 640x640 image size on VOC2007 test. The FPS is measured with batch size 1 on 3090 GPU from the model inference to the NMS operation.*
|
|
|
|
yjh0410