1. 程式人生 > >python中將資料增強

python中將資料增強

from keras.preprocessing.image import ImageDataGenerator
from keras.preprocessing import image
import matplotlib.pyplot as plt
import numpy as np
import pylab
img_generator = ImageDataGenerator(
    rotation_range = 90,
width_shift_range = 0.2,
height_shift_range = 0.2,
zoom_range = 0.3
)

# 匯入並顯示圖片
img_path = 
'G:\\100000.jpg' img = image.load_img(img_path) print(type(img)) # plt.imshow(img) # pylab.show() #img.show() # 將圖片轉為陣列 x = image.img_to_array(img) # 擴充一個維度 x = np.expand_dims(x, axis=0) # 生成圖片 gen = img_generator.flow(x, batch_size=2) # 顯示生成的圖片 plt.figure() for i in range(1): for j in range(1): x_batch =
next(gen) print(type(x_batch),x_batch.shape) idx = (1*i) + j plt.subplot(1, 1, idx+1) plt.imshow(x_batch[0]/255) pylab.show()