Guanghan / ROLO

ROLO is short for Recurrent YOLO, aimed at simultaneous object detection and tracking
Apache License 2.0
879 stars 278 forks source link

Suggesting different image printing for 3rd party script sort_yolo.py #35

Open arturioxas opened 6 years ago

arturioxas commented 6 years ago

Hello,

I tried to use your code for inspiration while doing tracking with different tool (YOLO+SORT) and realised matplotlib is painfully slow. I would suggest changing this part to use cv2. For me speed has gone from 5 fps to 10 fps for Full HD video. I could even try to make pull request if I could find some time in the future.