灰度图像预训练模型的处理

Question

我想使用一个新的网络，如VGG，ResNet，盗梦空间等进行传输学习。

问题是我的图像是灰度的(1通道)，因为上面提到的所有模型都是在ImageNet数据集(由RGB图像组成)上训练的。

解决方案之一是重复3次图像阵列，使其成为3通道。

这真的是唯一的解决办法吗？这是一个很好的解决办法吗？还有其他解决办法吗？

Answer 1

print(grayscale_batch.shape)  # (64, 224, 224)    

rgb_batch = np.repeat(grayscale_batch[..., np.newaxis], 3, -1)    
print(rgb_batch.shape)  # (64, 224, 224, 3)