【发布时间】:2020-01-05 06:09:10
【问题描述】:
我有一个 1920×1153×3 的 keras 张量,需要裁剪为 1920×1080。但是,创建裁剪图层并传递形状的差异会导致其尺寸错误。这种方法使张量的形状为 1920×1920。
代码:
def crop(x, size):
shape = K.int_shape(x)
shape_diff = shape[1] - size[1], shape[2] - size[2]
print(shape_diff)
y = tf.keras.layers.Cropping2D(shape_diff)(x)
return y
model.summary 和 shape_diff 的部分输出:
形状差异:
(0,72)
model.summary:
conv2d_transpose_4 (Conv2DTr (None, 1920, 1152, 3) 195
_________________________________________________________________
cropping2d (Cropping2D) (None, 1920, 1008, 3) 0
=================================================================
【问题讨论】:
标签: python arrays tensorflow keras crop