如何加载我们自己的数据集进行培训

时间:2019-07-29 20:05:16

标签: python tensorflow deep-learning

我想训练一个模型,该模型可以告诉我们任何图像的PM2.5(此值描述AQI)。为此,我使用CNN。我正在为此使用tensorflow。我是这个领域的新手。请告诉我我们如何上传自己的数据集,并分隔其名称和标签。图片名称的格式为“ imageName_tag”(例如ima01_23.4)

1 个答案:

答案 0 :(得分:1)

我认为我们需要有关“如何上传我们自己的数据集”的案例的更多信息。

但是,如果您的数据集在计算机上,并且您想通过python访问它,我邀请您看一下“ glob”和“ os”库。

要拆分名称(在您的情况下为“ imageName_tag”),可以使用:

string = "imageName_tag"
name, tag = string.split('_')

由于必须对所有数据进行处理,因此必须循环使用并将提取的信息存储在列表中。