如何提取标签

问题描述

这是数据示例： 100_0_0_20170112213500903.jpg.chip.jpg,100

这是python的代码，用于预处理数据集（因为标签将从图像名称中提取，如下代码所示：

textdata = textdata.map(lambda line: tf.strings.split(line,','))
data_image = textdata.map(lambda fnames: self.read(fnames[0]))

data_label = textdata.map(lambda classes: classes[1])
data_label = data_label.map(lambda classes: tf.strings.to_number(classes))

在我的情况下，我的数据集位于两个文件夹中，每个文件夹都带有标签

1. 火车

• 猫

• 狗

所以，请问如何更改此代码？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）