PRBonn / bonnetal

Bonnet and then some! Deep Learning Framework for various Image Recognition Tasks. Photogrammetry and Robotics Lab, University of Bonn
MIT License
233 stars 60 forks source link

FPS #17

Open sctrueew opened 4 years ago

sctrueew commented 4 years ago

Hello,

I'm using PyTorch GPU and "MobilenetsV2 ASPP Res - 512px" model with Python and C++ but I get ~94ms on my PC. Is it ok or not?

OS: Win 10 GPU: GTX 1080 Ti Video size: 2160x3840 Video FPS: 60

Thanks

tano297 commented 4 years ago

I don't have a 1080Ti to try it out right now, or a windows machine, but 10fps sounds a bit low to me. You should give the tensorrt model a try. You can convert it using the steps from the python part readme