【问题标题】:Training and validation samples are not what I have训练和验证样本不是我拥有的
【发布时间】: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


【解决方案1】:

问题是因为我正在阅读彩色图像(3 通道),而我的 inout 形状如下:

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

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2013-09-26
    • 1970-01-01
    • 2016-08-26
    • 1970-01-01
    • 2012-06-04
    • 2021-12-10
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多