如何修改GreyScale图像的Alexnet架构？

Question

我需要从alexnet架构中提取fc7中的功能。但我只有灰度图像。我怎样才能修改alexnet？

Answer 1

将预期的输入张量从3个通道重塑为1.您可以单独保留模型的其余部分。如果这过于适合您的数据集（您没有指定输入; AlexNet是为ImageNet构建的），那么请尝试将前两个卷积层的深度减少2倍。