enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 313 forks source link

batch inference is not supported yet? #67

Open nucthec-lyc opened 3 years ago

nucthec-lyc commented 3 years ago

thanks for your wonderful work! the batch inference now is not supported again? readme shows a n_max_batch parm. but it is not in class_detector.h of this version. please cheching ,thanks a lot.

enazoe commented 3 years ago

batch inference is supported now ,the max batch size is read from cfg file now. I will fix the readme sry