training_images, _ = next(test_dir.flow_from_directory(train_dir, target_size=(224, 224), batch_size=32))
plotImages(training_images[:10])

test_dir 应该是一个数据生成器,而不是一个可迭代对象,因此无法使用 next() 函数从中获取一批图像数据。使用上述代码可以获取一批图像数据并绘制前 10 张图像。

其中,train_dir 是训练集数据的文件夹路径,target_size 是图像的目标尺寸,batch_size 是批量获取的图像数量。请确保 train_dir 路径正确,并根据实际需求修改 target_size 和 batch_size 的值。


原文地址: https://www.cveoy.top/t/topic/pikj 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录