手寫體數字識別-建立訓練集

將黑白圖片轉成文本,就是讀取該圖片的像素,得到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:



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章