From c789e6da41c3e661bd9e89b934e5909fd6a3ff1d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=94=90=E7=92=9E?= <1039183305@qq.com> Date: Wed, 1 Feb 2023 08:19:07 +0000 Subject: [PATCH] update TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 唐璞 <1039183305@qq.com> --- .../tflib/ops/conv2d.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py b/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py index 3cec271ab..23f1a6ab5 100644 --- a/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py +++ b/TensorFlow/contrib/cv/Transferring-GAN_ID1252_for_Tensorflow/tflib/ops/conv2d.py @@ -131,15 +131,15 @@ def Conv2D(name, input_dim, output_dim, filter_size, inputs, he_init=True, mask_ if mask_type is not None: with tf.name_scope('filter_mask'): filters = filters * mask - + result = tf.transpose(result, perm=[0, 2, 3, 1]) result = tf.nn.conv2d( input=inputs, filter=filters, strides=[1, 1, stride, stride], padding='SAME', - data_format='NCHW' + data_format='NHWC' ) - + result = tf.transpose(result,perm=[0, 3, 1, 2]) if biases: _biases = lib.param( name+'.Biases', -- Gitee