taipingeric / yolo-v4-tf.keras

A simple tf.keras implementation of YOLO v4
MIT License
141 stars 78 forks source link

Convert Custom Yolov4 model #1

Open sharoseali opened 3 years ago

sharoseali commented 3 years ago

HI! I want to ask that I want to convert Yolov4 which has trained on 12 classes, hence I change some filter values. Will this repo will change my weights to the h5 Keras model successfully ??

taipingeric commented 3 years ago

Yes, remember to set the right class names file to build your model. You can check if you got a error message from here