python的数据增强利用opencv读取图片

# -*- coding: utf-8 -*-
__author__ = 'Administrator'

# import packages
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
import numpy as np
from keras.preprocessing import image
import matplotlib.pyplot as plt
import pylab
import cv2
datagen = ImageDataGenerator(
        width_shift_range=0.1,
        height_shift_range=0.1,
        shear_range=0.2,
        zoom_range=0.5,
        rotation_range=90,
        fill_mode='nearest')


##查询文件名
queryname=np.loadtxt('G:\\Holiday\\query_name.txt','str')

#训练集
X_train=np.zeros([500,224,224,3])
for i in range(10):
    if(queryname[i] != '130000.jpg'):
        img=cv2.resize(cv2.imread('G:\\Holiday\\jpg1\\jpg\\'+queryname[i]),(224,224),interpolation=cv2.INTER_LINEAR)
        img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
        img=np.expand_dims(img,axis=0)

    else:
        img=cv2.resize(cv2.imread('G:\\Holiday\\jpg1\\jpg\\' + queryname[i]), (224, 224),interpolation=cv2.INTER_LINEAR)
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        img=np.expand_dims(img, axis=0)


        #数据加强
    j = 0
    for batch in datagen.flow(img, batch_size=1, save_to_dir='G:\\Holiday\\数据增强', save_prefix=str(i),save_format='jpg'):

        j += 1
        if j > 20:  # 这个20指出要扩增多少个数据
            break  # otherwise the generator would loop indefinitely



# the .flow() command below generates batches of randomly transformed images
# and saves the results to the `preview/` directory