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.19k stars 315 forks source link

fix yolov4 broken detection #183

Closed marouf333 closed 1 year ago

marouf333 commented 1 year ago

1- fix missing cuda stream in calls to cuda kernels of mish, chunk and hardswish. 2- in parsing the config, trim the line before checking its empty, otherwise it could crash while checking an empty line for # symbol.