如何让tfds.folder\u dataset.ImageFolder调整图像大小?

2024-09-27 09:37:11 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经设置了一个图像目录,这样我就可以使用tfds.folder_dataset.ImageFolder,但是我在设置形状时遇到了一个问题。当我传递一个shape参数时,我得到一个错误:InvalidArgumentError: Incompatible shapes at component 0: expected [256,256,3] but got [4000,4000,3].图像形状在我的目录中是4000,4000,3,但我希望我可以使用tfds.folder_dataset.ImageFolder将它们调整为256,256,3,因为它们被加载到模型中(就像你可以使用Keras一样)。但是它似乎传递了shape参数,应该说是什么而不是我想要的是什么。那么,我如何告诉tfds.folder_dataset.ImageFolder我希望它将我的图像调整为指定的形状呢

编辑:如果没有使用tfds.folder_dataset.ImageFolder的方法,那么首选的方法是什么?我可以编写一个解析器并使用tf.data,但是tfds.folder_dataset.ImageFolder已经完成了我想要的一切


Tags: 方法图像目录参数错误folderdatasetat

热门问题