search

yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
372 stars 114 forks source link

Did you test the GPU speed measures end-to-end time per image? #10

Closed mheriyanto closed 3 years ago

mheriyanto commented 3 years ago

I tested your code to detect object on a video using GPU. I used GTX 1050 Ti LP and I got 3 FPS. What is it corresponding with your test result? Thanks @yasenh

yasenh commented 3 years ago

Hi @mheriyanto, you could refer to this issue #3, and I tested it with a GeForce GTX 1070 GPU. So you can use $ nvidia-smi -l to make sure your GPU is utilized during the inference, 3 FPS sounds little bit slower than I expected even on a 1050 Ti GPU.

mheriyanto commented 3 years ago

Okay, Thanks @yasenh, I will close this issue and then refer to issue #3.