标签: tensorflow deep-learning
我正在尝试使用Tensorflow进行传输学习,使用预先训练的VGG16模型。
但是,在我的问题中对模型的输入是一个RGB图像,其中一个额外的通道用作二进制掩码。这与训练模型的原始输入(224x224 RGB图像)不同。
我认为在这种情况下仍然可以使用预训练模型。如何为第一个卷积层和额外通道之间的连接分配权重?转学是否仍然适用于这种情况?
谢谢!