python - 将我的图像矢量化、 reshape 和标准化为 mnist.train.images 等图像

Question

如何对图像进行矢量化、 reshape 和标准化，使其与 mnist.train.images 中的其中一张图像的大小矢量相同？到目前为止我已经尝试过以下方法但没有成功:

import os
import re
import numpy as np
import tensorflow as tf
from tensorflow.python.framework import ops
import os,sys
#import Image
from PIL import Image
from tensorflow.contrib.learn.python.learn.datasets.mnist import read_data_sets

# Load data and check the shape of the first mnist.train.images image 
data_dir = 'temp'
mnist = read_data_sets(data_dir)
print("tmnist.train.images[0].shape is")
print(mnist.train.images[0].shape) # the result is (784,)

def resize_image(image):
    img = Image.open(image)
    arr = np.array(img)
    #my mind is drawing a blank... additional code to make this work...
    return arr


resize_image("test.png")

Answer 1

以下应该有效:

def resize_image(image):
   img = Image.open(image)
   img = img.resize((28, 28))
   arr = np.array(img)

   #convert to gray scale
   if len(arr.shape) > 2:
     arr = np.mean(arr, 2)

   #flatten 
   arr = arr.flatten()
   return arr