用Caffe+python如何训练一个手写体数字数别模型
手写体数字识别
1.将train.csv数据转化为lmdb格式存储
用Panda读取数据,由于.csv内的数据是以字符串形式存储,因此在读取完所有数据后,还需要将读取的数据转换成uint8形式,此时刚好可以存放一个像素值。具体操作如下:
np.fromstring(data, dtype=np.uint8)
然后从数据中分别提取图像信息和对应的标签信息:
labels = data
原创
2017-06-22 21:09:49 ·
754 阅读 ·
0 评论