阿里云PAI读取图像

最近PAI免费，研究了一下，简直就是学生党的福音啊，希望将来收费的时候能便宜点。
FLAGS.buckets获取的就是数据源选择的文件夹，然后调用这个read_image_in_pai函数就能读取目录内的全部图像。

def read_image( filepath ):
    img_obj = file_io.read_file_to_string(filepath)
    file_io.write_string_to_file("temp.jpg", img_obj)
    img = scipy.ndimage.imread("temp.jpg", mode="RGB")
    return img

def read_image_in_pai(FLAGS):
    img = np.zeros([1264,128,48,3])
    dirname = os.path.join(FLAGS.buckets, "")
    files = tf.gfile.ListDirectory(dirname) 
    for i in range(len(files)) :
        imagepath = os.path.join(FLAGS.buckets, files[i])
        img[i] = read_image(imagepath)

parser = argparse.ArgumentParser()
#获得buckets路径
parser.add_argument('--buckets', type=str, default='',
                        help='input data path')
#获得checkpoint路径
parser.add_argument('--checkpointDir', type=str, default='',
                        help='output model path')
FLAGS, _ = parser.parse_known_args()