【手写数字图片预处理】用skimage将自己的手写数字转换为mnist数据集

处理思路:读入图片>>>转换为28*28的大小>>>进行二值化处理>>>转换为向量

from skimage import io,data,transform
import numpy as np
import matplotlib.pyplot as plt
#目标大小
MNIST_SIZE = 28
def translate(image_path):
    #读入图片并变成灰色
    img = io.imread(image_path, as_grey=True)
    #缩小到28*28
    translated_img = transform.resize(img, (MNIST_SIZE, MNIST_SIZE))
    #变成1*784的一维数组
    flatten_img = np.reshape(translated_img, 784)
    #mnist数据集中1代表黑，0代表白
    result = np.array([1 - flatten_img])
    #返回该图的所代表的向量
    return result