Fix bug with multiple class indexes

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 bug with multiple class indexes #21

Closed Nuzhny007 closed 4 years ago

Nuzhny007 commented 4 years ago

Fix for this issue: https://github.com/enazoe/yolo-tensorrt/issues/20