WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

Speed #81

Open augmentedstartups opened 2 years ago

augmentedstartups commented 2 years ago

Currently I am getting only 10-15 FPS on my webcam 1280x960 resolution - on GTX 1080Ti On full HD its only at 6-8FPS

How do I run YOLOR at a faster framerate?

WongKinYiu commented 2 years ago

I have not tested on 1080ti. The bottleneck may be the frame rate of the webcam or usb port, you could use low resolution input to test if the bottleneck is it. If the bottleneck is model and you do not want to change model, I suggest you to use tensorrt.

augmentedstartups commented 2 years ago

The performance is the same on prerecorded video as well.