search

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
Apache License 2.0
190 stars 46 forks source link

Pytorch gives better FPS on videos #9

Closed makaveli10 closed 4 years ago

makaveli10 commented 4 years ago

I tested on one video and pytorch without NMS gives a lot better FPS that TRT inference I have shared the code that I used to inference TRT and for PyTorch I used detect.py in ultralytics/yolov5 repo to compute FPS (Both calculated without NMS). Image Size -> 640 * 640 PyTorch(without NMS) -> 125+ TRT (without NMS) -> 105

makaveli10 commented 4 years ago

Solved. Actually, I as using FP32 on TRT and FP16 on pytorch so it was giving better performance.