search

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

How can I change the load_weights function #376

Open KuoEuran opened 3 years ago

KuoEuran commented 3 years ago

Hi, I am training a yolov4-tiny weights, but I change convolutional layer with depthwise convolutional layer, how can I change the load weight function in utils.py to fit my weights. It shows the error image I have no idea where the name of conv%d define, and I can't change to depthwise convolutional layer. Please give me some advice, thanks.

sutejoramadhan commented 3 years ago

Hi @KuoEuran may this comment can help you https://github.com/hunglc007/tensorflow-yolov4-tflite/issues/362#issuecomment-881712107

KuoEuran commented 3 years ago

@sutejoramadhan Hi, thank you for your replying. I think the solution is not that. My question is that I want to use https://www.tensorflow.org/api_docs/python/tf/keras/layers/SeparableConv2D instead of normal convolutional, but I don't know how to modify the load weight function. Do you know the solution?

KuoEuran commented 3 years ago

@hunglc007 Hi, do you know about how to modify the separable convolutional weights? tks