Question

您好，我想知道是否存在一种方法，可以使用层将张量除以模型中keras中的常数，而不是例如使用ImageDataGenerator。

sequence = [
    keras.Input(shape=(224,224,3)),
    <------ DIVIDES ALL THE INPUTS BY 255
    keras.layers.Dense(32, activation="softmax"),
]
model = keras.Sequential(sequence)

Answer 1

您可以使用keras.Lambda层。

sequence = [
    Input(shape=(224,224,3)),
    Lambda(lambda x: x/255.0),
    Dense(32, activation="softmax"),
]

这实际上产生以下输出形状。 (None, 224, 224, 32)。例如，如果您要进行图像分类问题，这不是您想要的形状。然后，您只需要(None, 32)。根据您要解决的问题，上述形状可能会很好。但是只是指出以防万一这是一个错误。您可以按如下方式使用Flatten层。

sequence = [
    Input(shape=(224,224,3)),
    Lambda(lambda x: x/255.0),
    Flatten(),
    Dense(32, activation="softmax"),
]

keras将输入除以常量而不使用ImageDataGenerator

1 个答案: