lingtengqiu / Yolo_Nano

Pytorch implementation of yolo_Nano for pedestrian detection
140 stars 15 forks source link

yolo_nano的运行速度 #2

Open young-dl opened 4 years ago

young-dl commented 4 years ago

你好,请教一下,你是怎么测的yolo_nano的运行速度比yolov3-tiny快的,我测的yolo_nano的运行速度比yolov3-tiny慢4倍。

lingtengqiu commented 4 years ago

我们刚刚重新check 了这个时间,慢4倍是不太快可能的,正确的时间是去除掉dataloder 的时间,这个和核速有关,只检测 inference的时间, 因为这里的NMS 可能有点问题,当det 过多的时候时间会减慢,具体的时间速度,我们测试是 优于 yolo-v3,略低于 yolo-tiny。 感谢的你的反馈。

young-dl commented 4 years ago

我们刚刚重新check 了这个时间,慢4倍是不太快可能的,正确的时间是去除掉dataloder 的时间,这个和核速有关,只检测 inference的时间, 因为这里的NMS 可能有点问题,当det 过多的时候时间会减慢,具体的时间速度,我们测试是 优于 yolo-v3,略低于 yolo-tiny。 感谢的你的反馈。

感谢你的回答,我用你代码中的 inference.py 这个代码测的,只是测试模型前传的时间,在cpu上测得时间分别如下:对于一张图,yolo-nano的时间是0.3s,yolov3-tiny的时间是0.08s,而且yolo-nano每层的时间我也算了一下,layer0, layer1, layer2, layer3, layer4 和 layer5 的时间分别是0.01,0.09,0.03,0.07,0.03 和 0.01。 所以我还是不太明白这个测试的速度差距是在哪块?希望能够解答一下,谢谢!

BokyLiu commented 4 years ago

是的,速度偏慢,我把算法部署在rk3399上只能跑到25fps