Inference on a single image on an 1080 TI GPU is 2 seconds?!

xiaochus / YOLOv3

Keras implementation of yolo v3 object detection.

MIT License

603 stars 263 forks source link

Inference on a single image on an 1080 TI GPU is 2 seconds?! #9

Closed ghost closed 5 years ago

ghost commented 6 years ago

How could this be possible? I am using the demo.py script.

kevintsok commented 5 years ago

Yes. I have this problem as well. With my trained weight, inference with darknet takes about 0.3 secs, but it takes 2.6 sec for the forward propagation of keras model.