我已经设置了一个图像目录,这样我就可以使用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
已经完成了我想要的一切
目前没有回答
相关问题 更多 >
编程相关推荐