Convert separable convolutional layer weight to tflite

hunglc007 / tensorflow-yolov4-tflite

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

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.23k stars 1.24k forks source link

Convert separable convolutional layer weight to tflite #418

Open KuoEuran opened 2 years ago

KuoEuran commented 2 years ago

Hi, I have trained a Yolo model based on yolov4-tiny include separable convolutional layer, but the conversion is not successful, I think the problem is the conv_layer.set_weights in utils.py. Does anyone solve the problem? Please give me some advice of the problem, thanks.