theAIGuysCode / yolov4-deepsort

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.
GNU General Public License v3.0
1.31k stars 745 forks source link

yolov4 with different nr of classes #105

Open serno opened 2 years ago

serno commented 2 years ago

Hello, First of all thank you for awesome work. I have a yolov4 network trained for 3 classes that I would like to use. I changed the C.YOLO.CLASSES to my own names file, but I guess I need to change the nr. of conv. filters in the network (same as when I trained it for 3 classes) but I haven't found this in your version. How can I do this here, or is this already built into the code?