训练集:VOC2007+12训练集
代码 | 一个epoch所用时间 | ||
YOLOv4 | 10:00 | 3.52it/s | |
YOLOv3 | 7:40 | 4.64it/s | |
SSD+backbone=vgg | 3:20 | 5.23it/s | |
SSD+backbone=mobilenetv2 | 1:44 | 9.86it/s |
2G显存:SSD、YOLOV4-TINY 4G显存:YOLOV3 6G显存:YOLOV4、Retinanet、M2det、Efficientdet、Faster RCNN等 8G+显存:随便选吧
训练集:VOC2007+12训练集
代码 | 一个epoch所用时间 | ||
YOLOv4 | 10:00 | 3.52it/s | |
YOLOv3 | 7:40 | 4.64it/s | |
SSD+backbone=vgg | 3:20 | 5.23it/s | |
SSD+backbone=mobilenetv2 | 1:44 | 9.86it/s |
2G显存:SSD、YOLOV4-TINY 4G显存:YOLOV3 6G显存:YOLOV4、Retinanet、M2det、Efficientdet、Faster RCNN等 8G+显存:随便选吧