回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我使用下面的代码读取文件夹中的所有图像,并使用它们进行图像增强。<code>load_images()</code>函数将所有图像读取为numpy数组,但当我在代码的第二部分中使用此函数作为图像增强的输入时,会出现错误(<em>使用序列设置数组元素</em>)。感谢您的帮助。你知道吗</p>
<pre><code>from tensorflow.keras.preprocessing.image import ImageDataGenerator
from matplotlib.pyplot import imread, imshow, subplots, show
from matplotlib import pyplot as plt
import matplotlib.image as mpimg
import os
image_path = '/path/to/images/'
def load_images(image_path):
imagees = []
for filename in os.listdir(image_path):
img = mpimg.imread(os.path.join(image_path, filename))
if img is not None:
imagees.append(img)
return imagees
datagen = ImageDataGenerator(rotation_range=90, width_shift_range=0.3,
height_shift_range=0.3,shear_range=45.0, brightness_range=(0.1, 0.9),
zoom_range=[0.5, 1.5],channel_shift_range = 150.0, horizontal_flip=True, vertical_flip=True)
images = load_images(image_path)
images = image.reshape((1, image.shape[0], image.shape[1], image.shape[2]))
save_here = '/path/to/images/'
datagen.fit(images)
for x, val in zip(datagen.flow(images,
save_to_dir=save_here,
save_prefix='aug',
save_format='png'),range(36)):
pass
</code></pre>