tensorflow-yolov4-tflite icon indicating copy to clipboard operation
tensorflow-yolov4-tflite copied to clipboard

tensorflow depthwise convolution weight set weight problem

Open KuoEuran opened this issue 3 years ago • 0 comments

Hi, I have trained yolov4-tiny with depthwise convolutional layer and got a weight file. My darknet depthwise cfg is like image , and I use Netron to look my cfg file like image I changed my normal convolution in backbone.py, and I used tf.keras.layers.SeparableConv2D in common.py. image However, I got a bad result about the set_weights in utils.py, it showed get_weights (3, 3, 2048, 1) in tf.keras.layers.SeparableConv2D. I need (3, 3, 8, 256) in netron cfg. image

Does anyone solve the issue like this? Please give me some advice, thanks.

KuoEuran avatar Jan 24 '22 03:01 KuoEuran