如何制作我自己的mnist数据集?

时间:2017-03-09 18:07:55

标签: python tensorflow

Tensorflow mnist下载mnist数据集的英文数字,​​我正在使用mnist处理阿拉伯数字,我有JPG!

如何将JPG转换为我自己的“t10k-images-idx3-ubyte.gz”和“t10k-lables-idx1-ubyte.gz”?

请帮助我,谢谢。

1 个答案:

答案 0 :(得分:2)

您不需要将它们转换为字节和tar格式。您可以将它们输入占位符。

img = tf.placeholder(tf.float32, shape=(None))

...

digit = scipy.ndimage.imread("arabic_digit.jpg")
feed_dict = {img:digit}
sess.run(model, feed_dict=feed_dict)

如果您有多个数字,只需将它们加载到numpy数组中,然后再将其输入到Feed dict中。

digits = []
for digit_location in digits_list:
    digits.append(scipy.ndimage.imread(digit_location))
digits = np.asarray(digits)
feed_dict = {img:digits}
sess.run(model, feed_dict=feed_dict)