在 CNN 中为训练数据分配标签

Question

我是 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...

我知道我需要循环并将类号分配给每个训练图像。但不知道如何，我在网上查过，但数据格式不同。

Answer 1

首先读取.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]

然后您可以读取数据并将其提供给您的模型。