when convert yolov4 tiny model to tf file happen error

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

MIT License

2.23k stars 1.24k forks source link

Open ericosmic opened 3 years ago

ericosmic commented 3 years ago

I training my yolov4 for custom data based on alexeyab rep. I try convert weight to tf file by use save_model.py , but it give a error :

I tried difference input size , but it seems not work.

jaehwlee commented 3 years ago

Did you check core/config.py? Replace coco.names to your labels. It works to me.

HosseinMohammadii commented 2 years ago