tensorflow-yolov4-tflite
tensorflow-yolov4-tflite copied to clipboard
tensorflow depthwise convolution weight set weight problem
Hi, I have trained yolov4-tiny with depthwise convolutional layer and got a weight file. My darknet depthwise cfg is like
, and I use Netron to look my cfg file like
I changed my normal convolution in backbone.py, and I used tf.keras.layers.SeparableConv2D in common.py.
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.

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