Closed varunvv closed 2 years ago
I am getting error while converting yolo-v4 model trained for detecting single class objects. What modification should I make in order to use a custom trained yolo-v4 model.
Generally you need to change all settings in the following lines to your model:
https://github.com/isarsoft/yolov4-triton-tensorrt/blob/543fde846b2751d6ab394339e005e2754de22972/networks/yolov4.h#L14-L31
I am getting error while converting yolo-v4 model trained for detecting single class objects. What modification should I make in order to use a custom trained yolo-v4 model.