将图像转换为MNIST格式时出现unsigned int溢出错误

Question

我是使用张量流的深度学习的新手。 我想创建自己的模型来预测在灰度上构建的自定义图像。 但是我唯一知道的是使用张量流的MNIST示例。

因此，我使用了this repo中的转换模块，但是发生了这样的错误。

要转换的图像被构造为80,680个训练图像，20,170个测试图像。

我真的不知道为什么会发生此错误。 请帮我。

Answer 1

您所引用的脚本未正确设置MNIST格式的标头。此问题在上一个Github问题中得到了解决，此问题已被删除，但我的modification：

header = array('B')
header.extend([0,0,8,1,0,0])
header.append(int('0x'+hexval[2:][:2],16))
header.append(int('0x'+hexval[2:][2:],16))

到

header = array('B')
header.extend([0,0,8,1])
header.append(int('0x'+hexval[2:][:2],16))
header.append(int('0x'+hexval[4:][:2],16))
header.append(int('0x'+hexval[6:][:2],16))
header.append(int('0x'+hexval[8:][:2],16))

应该使它工作。希望这会有所帮助！