【发布时间】:2018-03-30 05:15:57
【问题描述】:
我遇到了一个奇怪的问题。我正在尝试使用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
我确认我有正确数量的图像,但不知道为什么我在训练和验证中会有这么多的图像。
任何想法可能是什么问题?
谢谢。
【问题讨论】:
-
train_data到底是什么,你确定你的索引正确,而不是在一些奇怪的轴上导致numpy自动重塑你的数组吗?
标签: python tensorflow tflearn