Python学习：图片数据批量生成标签

最新推荐文章于 2024-07-06 03:33:12 发布

AI研习图书馆

最新推荐文章于 2024-07-06 03:33:12 发布

阅读量2k

点赞数 1

分类专栏：常用代码文章标签：数据预处理

本文链接：https://blog.csdn.net/weixin_44222014/article/details/103236784

版权

常用代码专栏收录该内容

19 篇文章 3 订阅

订阅专栏

个人微信公众号：AI研习图书馆，欢迎关注~

深度学习知识及资源分享，学习交流，共同进步~

1. 介绍

Python学习：对文件夹下，所有图片数据，提取名称及类别，批量生成标签文件txt

2. 程序设计

# coding=UTF-8

import os
import sys

path="E:/test1"
filenames=os.listdir(path) #读取path内所有文件名返回列表
i=0
with open("E:/test1/train_color.txt","w") as f:
        for filename in filenames:
                sub_path=path+"/"+filename+"/"
                pic=os.listdir(sub_path)
                print(filename+','+str(i))
                for p in pic:
                        f.write('zhifayi/'+filename+'/'+p+' '+str(i)+'\n')
                i=i+1