loops - 数据增强仅增强目录中的 1 个图像

Question

我试图用来从我的目录中增加 15 个图像的代码，但是当我运行代码时，它只从目录中选择 1 个图像来增加。我想增加所有 15 个图像。谁能帮我？

import keras
import cv2
import os
import glob
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
datagen = ImageDataGenerator(rotation_range =15, 
                         width_shift_range = 0.2, 
                         height_shift_range = 0.2,  
                         rescale=1./255, 
                         zoom_range=0.2, 
                         horizontal_flip = True, 
                         fill_mode = 'constant', 
                         data_format='channels_last', 
                         ) 


img_dir = "C:/Users/NAME/Desktop/School" #Enter Directory of all images 
data_path = os.path.join(img_dir,'*g')
files = glob.glob(data_path)
data = []
for f1 in files:
    img = cv2.imread(f1)
    data.append(img)

x = img_to_array(img)
x = x.reshape((1,) + x.shape)

i = 0
path, dirs, files = next(os.walk("C:/Users/NAME/Desktop/School/"))
file_count = len(files) #to find number of files in folder

for data in datagen.flow (x, batch_size=1, save_to_dir =r'preview',save_prefix="School",save_format='png'):
    i+=1
    if i==5:
      break

标签： loopsdata-sciencedata-augmentation