coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本,以适应不同的需求。
805 stars 214 forks source link

图像被缩放为256x416 #79

Open seek-ing opened 4 years ago

seek-ing commented 4 years ago

我在训练自己的数据时mAP为70%左右,按理来说检测效果应该不错,但是在检测的时候出现:

video 1/1 (1452/67673) /home/yhy/YOLOv3_pruning/ch03.mp4: 256x416 Done.

也就是图像被莫名的缩放成256x416. 修改detect.py第10行img_size = (416, 416)也没有效果。 mAP 70%,但是检测效果非常差。

然后我用 ultralytics /yolov3的工程加载你的工程训练好的权重做检测,图像还是被缩放成了 256x416,所以应该是训练时候的问题。

请问大家都是怎么解决这个问题的?

seek-ing commented 4 years ago

发现并不是256x416的问题,在训练的时候,训练参数最好跟ultralytics /yolov3的参数一致