我想使用预训练的 Resnet 50 作为 Unet 模型的主干。但问题是 resnet 50 期望图像的大小为 197 x 197 3D 通道，但我的图像是 128 X 128 x 1D 通道。是否有任何 keras 或 tensorflow 实现。
Resnet 50 作为 Unet 的主干
数据挖掘
喀拉斯
张量流
计算机视觉
2022-02-13 01:15:38
1个回答
当您以这种方式导入 ResNet50 时：
image_input=Input(shape=(512, 512, 3))
model = ResNet50(input_tensor=image_input, include_top=False)
之后，您可以使用所需的 image_size