首页 > 解决方案 > Keras 模型接受的宽度和高度

问题描述

我有一个宽度和高度的 rgb 图像数据集:291x193

我正在使用几种模型,例如 inception 和 effecientnet

我用这个维度可以吗??尺寸应该是奇数还是永远,或者没关系?

提前致谢!!

标签: pythonkerasdeep-learning

解决方案


由于您已在下面标记了此内容,因此keras我假设您正在使用 InceptionV3 函数tf.keras

include_top该函数有一个可选的形状元组,只有在is时才能指定False(否则输入形状必须是 (299, 299, 3) 遵循channels_last data格式或 (3, 299, 299) 具有channels_first数据格式。它应该正好有 3 个输入通道,宽度和高度不小于75。另外,如果input_tensor没有提供,您可以指定大小。您可以在input_shape元组中添加您需要的尺寸。

例如:

tf.keras.applications.InceptionV3(
    include_top=False,
    weights="imagenet",
    input_tensor=None,
    input_shape=(100, 1000, 3),
    pooling=None,
    classes=1000,
    classifier_activation="softmax",
)

推荐阅读