TF-deformable-conv
TF-deformable-conv copied to clipboard
deform_conv_2d() meet the error
when i build my own model,in this step:
i meet the errror:
look forward to hearing from you.
@Zardinality
you feed a NHWC format data into the layer of NCHW. its not related to this repo
i use this img = tf.transpose(img, [0, 3, 1, 2]) to change to the NCHW,but i got another errror:
@Zardinality
Before using deform_conv_2d(),convert the input format to NCHW:tf.transpose(img, [0, 3, 1, 2]); And convert the function output to NHWC:tf.transpose(img, [0, 2, 3, 1]).
I meet the same issue, and I found the arg data_format
of deform_conv_op
is useless.
Only make sure input a tensor with NCHW
can the op data_format
work.