我正在绘制我的图像生成器生成的图像。到目前为止，这是我给生成器的数据的代码：
train_img_gen = train_img_data_gen.flow_from_directory(os.path.join(training_dir, 'images'),
target_size=(img_h, img_w),
batch_size=bs,
class_mode=None, # Because we have no class subfolders in this case
shuffle=True,
interpolation='bilinear',
seed=SEED)
#edited part following the already existing answer on stackoverflow
x_batch, y_batch = next(train_img_gen)
for i in range (0,32):
image = x_batch[i]
plt.imshow(image.transpose(2,1,0))
plt.show()我跟着这个问题：Keras images，但没有任何成功。
例如，我如何绘制由imageGenerator生成的前n个图像？
编辑：
我添加了上面提到的问题中使用的代码，但我得到了这个错误：
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-7-1a18ce1c1a76> in <module>
54 valid_gen = zip(valid_img_gen, valid_mask_gen)
55
---> 56 x_batch, y_batch = next(train_img_gen)
57 for i in range (0,32):
58 image = x_batch[i]
ValueError: too many values to unpack (expected 2)
2条答案
按热度按时间pgky5nke1#
最后我解决了这个问题，这是一个关于尺寸的问题。
工作代码为：
生成器为每次迭代返回形状为
(4,256,256,3)的矩阵，这意味着我们有4个大小为256x256的图像和3个通道（RGB）。ImageDataGenerator一次处理4个图像的“块”（至少在这种情况下，我没有关于它每次加载多少图像的官方参考），因为它的主要目的是在训练模型时动态加载图像（避免在内存中预加载大量数据）。umuewwlo2#