之前一直是想解决从规定的文件名列表里面读取图片并且放入TF训练的问题,stackoverflow上面找到了这个答案:http://stackoverflow.com/questions/37450620/tensorflow-image-classification,虽然并非是我最想达到的效果,但毕竟是一个有用的蓝本,在之后以这个为基础再进行改进吧,所以想把这一段代码深入理解一下。
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import time
import math
import numpy
import numpy as np
import random
from PIL import Image
使用IPL库来读取图片,其实在这个框架里面就算使用pyplot.iamge来读取图片也没有什么问题
我对比了一些开源库对于读取图像这个问题的解决方案,能够跑的动的方案其实效果都不是非常理想。比如可以使用opencv来做读图图像的接口,YOLO-py中就是这么做的,大致代码如下:
import cv2
img = cv2.imread("xxx.jpg")
#you can do some cv2 process
img_arrayy = np.array(img).resize(imagePixes,1)
#do some tensorflow process
feeddict = {placeholder:im_array}
但是这些方法依然存在着内存消耗过大的问题&