Tensoflow 把自己的图片生成向量

在Python2下运行的代码

我先把工程目录截图放上来：

import os
import numpy as np
import cv2


def imgTodata(path, imgCount = 128, weight = 1960, height = 960, channel = 3):

pathDir = list(os.listdir(path))

# [pic, weight, height, channel]
imgs = np.zeros((imgCount, weight, height, channel))

for id, pic in enumerate(pathDir):
img = cv2.imread(str(path + pic))
# cv2.namedWindow('test')
# cv2.imshow('test', img)
# cv2.waitKey(0)
# print 'The ', id, ' picture shape: ', img.shape
newImg = cv2.resize(img, (int(height) ,int(weight)), interpolation=cv2.INTER_CUBIC)
# print 'The ', id, ' picture shape: ', newImg.shape

imgs[id] = newImg
# print id, ' is success!'

# print imgs.shape

for i in xrange(2):
img = imgs[i]
# print img.shape
return imgs

def save(path, data):
np.save(path, data)

def load(path):
data = np.load(path)
return data

if __name__ == '__main__':
path = 'data/'
data = imgTodata(path,2, 1280, 960, 3)
print data[0]
print data.shape
save_path = 'npy_data/test.npy'
save(save_path, data)
data_ = load(save_path)
print data_.shape

只要修改对应的参数，就可以把自己图片当成向量传给Tensorflow网络。