CaoWGG / TensorRT-CenterNet

tensorrt5 , centernet , centerface, deform conv, int8
767 stars 158 forks source link

Hao to deal with batch images #61

Open huituolj opened 4 years ago

huituolj commented 4 years ago

Thank you for your excellent work! I want to deal with batch(batch_size=3) images in the do_inference, where the do_inference func support? I have generate ctdet onnx and engine file(input_shape:3,3,512,512) successfully.