TrojanXu / yolov5-tensorrt

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

Num_det always zero #50

Open tuananhroman opened 3 years ago

tuananhroman commented 3 years ago

Env

Hey everybody,

I'm currently working on a panorama live tracking application and intended on speeding up the inference with TensorRT. For engine generation, I'm using the exact same code as in this repo. Subsequent inference always returns zero though, as num_det equals zero on every instance even though there is some reasonable result in the outputs array for outputs[0].

image image

How do I need to change line 165 in order to retrieve the true number of detections found? Or is there a fundamental error in the respective engine?

Thanks a lot!