search

thaitc-hust / yolov9-tensorrt

Yolov9 TensorRT Inference Python backend
29 stars 5 forks source link

batch_size > 1 #7

Closed UcanYusuf closed 6 months ago

UcanYusuf commented 6 months ago

Hello, how can I increase batch size? As I understand default batch_size=1.

thaitc-hust commented 6 months ago

@UcanYusuf in https://github.com/thaitc-hust/yolov9-tensorrt/blob/0d8ce42e9e85bcdd8ca0fc090a90ff558ff3f28a/object_detector_trt_nms.py#L294C9-L294C12 inp is batch image, you can append other image to it