search

isarsoft / yolov4-triton-tensorrt

This repository deploys YOLOv4 as an optimized TensorRT engine to Triton Inference Server
http://www.isarsoft.com
Other
276 stars 63 forks source link

Can I use this repo to use custom trained yolo-v4 with single class #62

Closed varunvv closed 2 years ago

varunvv commented 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.

philipp-schmidt commented 2 years ago

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