def conv2d_fixed_padding(**kwargs):
"""conv2d with fixed_padding, based only on kernel_size."""
strides = kwargs["strides"]
if strides > 1:
kwargs["inputs"] = fixed_padding(kwargs["inputs"], kwargs["kernel_size"],
kwargs["data_format"])
defaults = {
"padding": ("SAME" if strides == 1 else "VALID"),
"use_bias": False,
"kernel_initializer": tf.variance_scaling_initializer(),
}
defaults.update(kwargs)
return tf.layers.conv2d(**defaults)
评论列表
文章目录