machine-learning - 如何修改Graynet图像的Alexnet体系结构？

我需要从alexnet体系结构的fc7中提取功能。但是我只有灰度图像。如何修改alexnet呢？

最佳答案

将预期的输入张量从3个通道重塑为1。您可以不考虑模型的其余部分。如果这过度适合您的数据集（您尚未指定输入； AlexNet是为ImageNet构建的），则尝试将前两个卷积层的深度减小2倍。