CaoWGG/TensorRT-CenterNet

Hao to deal with batch images

Opened this issue · 0 comments

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.