在 keras 中使用调整大小图层时出错

Error while using the Resizing layer in keras

提问人:Achintya 提问时间:11/14/2023 更新时间:11/14/2023 访问量:11

问:

我正在训练一个模型,其中我有一个不同大小的图像数据集,现在我正在使用模型内 keras 中的“调整大小”层调整所有图像的大小


model = Sequential()

model.add(Resizing(height=375, width=500, crop_to_aspect_ratio=True))
model.add(Rescaling(scale=1./255.))

model.add(
    Conv2D(40, (3, 3), activation='relu', input_shape=(375, 500, 3))
)
model.add(
    MaxPooling2D((2, 2))
)

model.add(Flatten())
model.add(Dense(300, activation='relu'))
model.add(Dense(100, activation='relu'))
model.add(Dense(30, activation='relu'))
model.add(Dense(1, activation='sigmoid'))

model.compile(loss = BinaryCrossentropy(), metrics=['accuracy'])

training = model.fit(train_images, train_labels, epochs=10, validation_data = (test_images, test_labels))

我尝试删除crop_to_aspect_ratio参数并将训练和测试图像标签转换为 numpy 数组,但没有任何效果

Keras 深度学习 图像大小调整 预处理器

评论

1赞 Dr. Snoopy 11/14/2023
您始终需要在问题中包含错误,我们无法猜测或假设您得到的错误。

答: 暂无答案