手写体数字识别-建立训练集

将黑白图片转成文本,就是读取该图片的像素,得到RGB值,如果为0就是黑色转成1,否则就是白色转成0。要提前下载pillow模块,直接cmd,里面输入pip install pillow 就可以了。Image.getpixel(width,height)就可以得到该点的RGB值,不要颠倒成Image.getpixel(height,width)。另外注意,fh要用‘a’追加的方式打开,用w的方法就会不停的覆盖,写不进去。还有,要注意循环嵌套,是height套weight,还是weight套height,这两种不同嵌套,获得的文本是有可能与原图不同角度。

旋转图像:out=img.rotate(45) #逆时针旋转45度

from PIL import Image
im=Image.open("D:/xx/xx/2.png")
fh=open("D:/xx/xx/2shibie.txt",'a')
#保存为新图片
#im.save("")
#获取图片宽、高
width=im.size[0]
height=im.size[1]
for i in range(0,height):
    for j in range(0,width):
        cl=im.getpixel((j,i))
        clall=cl[0]+cl[1]+cl[2]
        if(clall==0):
            fh.write("1")
        else:
            fh.write("0")
    fh.write("\n")

fh.close()

2.png:



阅读更多

没有更多推荐了,返回首页