how to supporrt dynamic batch thx~

Monday-Leo / YOLOv7_Tensorrt

A simple implementation of Tensorrt YOLOv7

108 stars 15 forks source link

how to supporrt dynamic batch thx~ #4

Open wojiaohumaocheng opened 2 years ago

Monday-Leo commented 2 years ago

I haven't tried dynamic batch yet, you can modify export_onnx.py to make the input batch dynamic and then export the tensorrt model. The default batch value is 1.