【PyTorch教程】制作数据集的标签（label）

关于制作数据集的label：

• 在P7视频的最后，写了这段，用来重新命名文件的label的
• img 和 label 的管理方法，有两种：
  1、用img所在文件夹的名称，作为label；
  2、img在一个文件夹，label在另一个文件夹，img文件夹中是图片，label文件夹中是txt文件，txt文件名跟对应img名称一致，而label，写在txt文件里面

可以运行的代码

# ！usr/bin/env python3
# -*- coding:utf-8 -*-
"""
author :24nemo
 date  :2021年07月12日
"""
import os

root_dir = "TuDui/src/dataset/train"  # train: path from contend root
target_dir = "ants_image"  # ants_image: copy file name

img_path = os.listdir(os.path.join(root_dir, target_dir))
label = target_dir.split('_')[0]
out_dir = "ants_label"
for i in img_path:
    file_name = i.split('.jpg')[0]
with open(os.path.join(root_dir, target_dir, "{}.txt".format(file_name)), "w") as f:
    f.write(label)

运行后的效果

• train 数据集中，img对应的label作为文件夹的目录存在，当前目录当中的数据内容，与文件夹名称保持一致