AastaNV / TRT_object_detection

Python sample for referencing object detection model with TensorRT
142 stars 27 forks source link

problem with batch size > 1 #23

Open PythonImageDeveloper opened 4 years ago

PythonImageDeveloper commented 4 years ago

Hi, Thank you for sharing the codes. I used jetpack 4.2.2 and opencv 3.4.6 and tensorflow 1.12.2. and follow your codes for converting the model to uff and tensorrt model. but the structural of your code don't support batch of image processing at the same time. if prediction time of one image takes 50 ms, the prediction time of batch of three image take 150 ms, why? How do I solve the problem?