tensorflow使用图像路径作为训练输入

问题描述 投票:0回答:1

运行model.fit时,有没有办法使用包含所有训练图像文件的相对路径的列表作为输入?

例如,如果数组类似于:

['images/00001.png', 'images/00002.png', ... 'images/9999.png']

如何将其用作训练模型的输入?所有图像都是灰度图像,我不想将图像转换为数组,因为这会导致内存错误。

tensorflow out-of-memory
1个回答
0
投票

尝试将图像存储在元组中。它与Tensorflow一起很好地工作,并在需要时可以保存大量数据。

© www.soinside.com 2019 - 2024. All rights reserved.