6

有什么方法可以将 ResNet50 的 ImageNet 权重用于我的具有形状图像 (224,224,4) 的项目?图像具有 R、G、B、Y 通道。

目前,我只是在使用

 model = ResNet50(include_top=True, weights=None, input_tensor=None, input_shape=input_shape, pooling=None, classes=num_classes)

现在,如果我需要使用 ImageNet 权重,我需要始终将类数设置为 1000。我尝试这样做,然后弹出最后一层,并添加我自己的 Dense(num_classes) 层。但是,现在,频道的数量是一个问题。

任何人都可以建议一种在使用 ImageNet 权重时在模型中容纳 4 个通道的方法吗?

4

0 回答 0