我正在绘制我的图像生成器生成的图像。到目前为止,这是我给生成器的数据的代码:
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#