Question

如何对文件夹中存储的多个图像应用Keras图像增强？

P.S：我为单个图像尝试了以下代码，效果很好。

有人可以帮我解决多张图片吗？

enter code here

from keras.preprocessing import image

import matplotlib.pyplot as plt
import cv2

from keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(
            rotation_range = 30, 
            width_shift_range = 0.2,
            height_shift_range = 0.2, 
            shear_range = 0.2, 
            zoom_range = 0.2,
            horizontal_flip = 0.2,
            fill_mode = "nearest")

for img in glob.glob("Images/*/*.jpg"):
    cv_img = cv2.imread(img)
    cv_resize = cv2.resize(cv_img,(200,200))
    cv_norm_img = cv_resize/255.0
    break


cv_norm_img = np.array(cv_norm_img)

input_batch = cv_norm_img.reshape((1,*cv_norm_img.shape))

i = 0

for output_batch in datagen.flow(input_batch,batch_size=1):
    plt.figure()
    imgplot = plt.imshow(image.img_to_array(output_batch[0]))
    i+=1
    if i==10:
        break
    plt.axis('off')
    plt.show

Answer 1

最简单的方法是从目录使用ImageDataGenerator.flow。文档是在https://keras.io/preprocessing/image/。图片将是一个形状数组（batch_size，200、200、3）注意：每次您执行此操作时，都会在 save_to_dir，因此您可能不想包含该参数。您可以以img1 = images [0]，img2 = images [1]等访问单个图像如下设置发电机

git push origin

在Python中使用Keras进行图像增强

1 个答案: