如何解决在 CNN 中为训练数据分配标签
我是 kaggle 的新手,我想知道如何为我的测试数据分配标签。
我的火车和测试文件夹都有格式
1
2
3
etc...
train.txt 包含如下标签:
train/0.jpg 5
train/1.jpg 1
train/2.jpg 10
train/3.jpg 2
train/4.jpg 22
train/5.jpg 3
etc...
我知道我需要循环并将类号分配给每个训练图像。但不确定如何,我在网上查看过,但数据格式不同。
解决方法
首先读取.txt文件并制作路径列表和标签:
with open('train.txt','r') as f:
path_label = f.readlines() # returns a list
# seperate paths and labels
paths = [i.strip().split()[0] for i in path_label]
labels = [i.strip().split()[1] for i in path_label]
然后您可以读取数据并将其提供给您的模型。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。