我有一个包含灰度图像的数据集,我想在它们上训练一个最先进的CNN.我非常想调整一个预先训练好的模型(就像这里的模型一样).
问题在于我几乎所有可以找到权重的模型都在ImageNet数据集上进行了训练,该数据集包含RGB图像.
我不能使用其中一个模型,因为他们的输入层需要一批形状(batch_size, height, width, 3)或(64, 224, 224, 3)在我的情况下,但我的图像批次是(64, 224, 224).
有什么方法可以使用其中一种型号吗？我已经考虑过在加载权重并添加自己的权重后丢弃输入层(就像我们为顶层做的那样).这种方法是否正确？