Question

我有一个奇怪的问题。我正在尝试使用376图像训练卷积神经网络（CNN）。我已经设置了大约80％的图像用于培训，大约20％用于验证，如下所示：

train = train_data[:300]
test = train_data[300:]

但是当我运行该程序时，我得到以下内容：

Training samples: 900
Validation samples: 228

导致以下错误：

IndexError: index 525 is out of bounds for axis 0 with size 300

我确认我的图像数量正确，并且不确定为什么我在训练和验证中有这么多额外的图像。

任何想法可能是什么问题？

感谢。

Answer 1

问题是因为我正在阅读彩色图像（3通道），而我的外形如下：

convnet = input_data(shape=[None, IMG_SIZE, IMG_SIZE, 1], name='input')

所以，我只需要将1替换为3。