【发布时间】:2020-07-13 10:08:24
【问题描述】:
使用 Keras API 训练卷积神经网络,在使用彩色 png 图像(输入大小(高度、宽度、4))进行训练时,我通常使用 2D 卷积层。但是,我现在希望使用灰度 png 图像训练网络,我想知道最好的方法是什么?
我想出了三种可能的方法:
- 像输入彩色图像一样输入图像。
- 只输入图片的第一个通道,这样输入的尺寸就是(高,宽,1)。
- 使用图像第一个通道的一维卷积,使输入大小为(高度、宽度)。
由于我的 png 格式的灰度图像在所有 RGBA 通道中具有相同的值,我认为我可以通过仅使用图像的第一个颜色通道进行训练和测试来减少计算时间,同时实现相同的测试精度因为数据本质上是相同的。然而事实并非如此。
使用方法一,我能够达到 91.95% 的测试准确率,每个 epoch 大约需要 3 秒的训练时间。
使用方法二,准确率为 89.66%,每个 epoch 约 2 秒。
使用方法三,准确率为 86.21%,每个 epoch
所有网络都使用相同的架构、内核大小和池大小进行训练,所以我想知道是什么导致了准确性的差异,我应该最信任哪一个?
【问题讨论】:
标签: machine-learning deep-learning neural-network computer-vision conv-neural-network