学习笔记-----制作图片分类标签

机器学习图片分类数据集标签制作步骤

一、准备好数据集

在项目下添加datasets目录，e0和e1文件夹内分别是需要打上标签为0和1的图片。

二、打标签的代码

import os

def generate(dir, label):
    files = os.listdir(dir)  # os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。
    files.sort()  # 对文件或文件夹进行排序
    print('****************')
    print('input :', dir)
    print('start...')
    listText = open(r'D:\Code\pycharm\pytorch\makelabel\datasets\all_list.txt',
                    'a+')  # 创建并打开一个txt文件，a+表示打开一个文件并追加内容

    for file in files:  # 遍历文件夹中的文件
        fileType = os.path.split(file)  # os.path.split（）返回文件的路径和文件名，【0】为路径，【1】为文件名
        if fileType[1] == '.txt':  # 若文件名的后缀为txt,则继续遍历循环，否则退出循环
            continue
        name = folder + '/' + file + ' ' + str(int(label)) + '\n'  # name 为文件路径和文件名+空格+label+换行
        listText.write(name)  # 在创建的txt文件中写入name
    listText.close()  # 关闭txt文件
    print('down!')
    print('****************')


outer_path = r'D:\Code\pycharm\pytorch\makelabel\datasets\images'  # 这里是你的图片路径

if __name__ == '__main__':  # 主函数
    i = 0
    folderlist = os.listdir(outer_path)  # 列举文件夹
    for folder in folderlist:  # 遍历文件夹中的文件夹(若engagement文件夹中存在txt或py文件，则后面会报错）
        generate(os.path.join(outer_path, folder), i)  # 调用generate函数，函数中的参数为：（图片路径+文件夹名，标签号）
        i += 1

其中all_list.txt文件不需要自己创建，会自动生成。

三、运行结果

 第一列：文件夹/图片名     第二列：所属的类别

该文件的目录：D:\Code\pycharm\pytorch\makelabel\datasets\all_list.txt（在datasets下）

参考文章链接：制作数据集（二）--为图片数据集打上标签并保存为txt文件_困坤的小菜鼠的博客-CSDN博客