search

linhaoqi027 / yolov5_openvino_sdk

an SDK about how to use openvino model transformed from yolov5
36 stars 11 forks source link

yolov5s.pt convert to opnevino test time #4

Open Belinda-great opened 3 years ago

Belinda-great commented 3 years ago

I use yolov5s, v3, img size is 640*640, one thread, test inference time, test openvino(2020R4) time is 1.28s, and pytorch one thread time is 1.2s, it's too slow. platform: E5-2699 V3 @2.30GHz. Is there anything wrong with my test? what time did you test?

linhaoqi027 commented 3 years ago

Did you use my code? If yes,it may caused by NMS. Beacuse NMS is not include in openvino.I convert the model output to torch then using NMS code in yolov5 official code. If you want shorten test time.You should do openvino in NMS. If you done this, please contract me to update this project