图片转为mnist
参考了别人的处理加上自己改进,
注意输入图片是黑底白字,并且要把大小给改好,28*28,不然就会报错。
手写体数字识别
特别给力的单张图片转换mnist
import os
from PIL import Image
from array import *
from random import shuffle
import sys
import cv2
def changeFile(fileName):
imageData = array('B')
FileList = []
# 读取图像,第二个参数是读取方式
img = cv2.imread(fileName, 1)
# 使用全局阈值,降噪
ret,th1 = cv2.threshold(img,127,255,cv2.THRESH_BINARY)
# 把opencv图像转化为PIL图像
im = Image.fromarray(cv2.cvtColor(th1,cv2.COLOR_BGR2RGB))
# 灰度化
im = im.convert('L'