深度学习，图⽚数据集⽣成标签⽂件列表，即.txt⽂件（lmdb ⽂件格式）
在深度学习中的图⽚训练时，需要将图⽚⽣成标签⽂件列表，
本次我是对于多分类问题，⼀共８个类别，python代码实现如下： #coding:utf-8
import os
def IsSubString(SubStrList,Str):
flag=True
for substr in SubStrList:
if not(substr in Str):
flag=False
return flag
#扫⾯⽂件
def GetFileList(FindPath,FlagStr=[]):
延时电路FileList=[]
FileNames=os.listdir(FindPath)
if len(FileNames)>0:
for fn in FileNames:
if len(FlagStr)>0:
农药渗透剂if IsSubString(FlagStr,fn):
fullfilename=os.path.join(FindPath,fn)
FileList.append(fullfilename)
else:
fullfilename=os.path.join(FindPath,fn)
FileList.append(fullfilename)
if len(FileList)>0:
FileList.sort()
水塔水位控制系统return FileList
train_txt=open('','w')
#制作标签数据，多标签数据，编号从0开始
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/cabbage')#.py⽂件⽬录下
for img in imgfile:
str1=img+' '+'0'+'\n' #⽤空格代替转义字符 \t
train_txt.writelines(str1)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/carrot')
干墙
for img in imgfile:
str2=img+' '+'1'+'\n'
train_txt.writelines(str2)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/cauliflower')
for img in imgfile:
str3=img+' '+'2'+'\n'
train_txt.writelines(str3)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/cucumber')
for img in imgfile:
str4=img+' '+'3'+'\n'
train_txt.writelines(str4)
train_txt.writelines(str4)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/eggplant')
for img in imgfile:
str5=img+' '+'4'+'\n'
train_txt.writelines(str5)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/green_pepper') for img in imgfile:
str6=img+' '+'5'+'\n'
train_txt.writelines(str6)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/potato')
for img in imgfile:
str7=img+' '+'6'+'\n'
train_txt.writelines(str7)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/pumpkin')
for img in imgfile:
str8=img+' '+'7'+'\n'
windows下刷biostrain_txt.writelines(str8)
imgfile=GetFileList('/home/aa/qxq/project/fruits/ncnn-master/data/train/tomato')
for img in imgfile:
str9=img+' '+'8'+'\n'
train_txt.writelines(str9)
#转换完成后，将.txt⽂档关闭
钛雷train_txt.close()
#测试集⽂件列表
#test_txt=open('','w')
#制作标签数据，如果是男的，标签设置为0，如果是⼥的标签为1
#imgfile=GetFileList('val/test_cat')#将数据集放在与.py⽂件相同⽬录下
#for img in imgfile:
# str3=img+' '+'1'+'\n'
# test_txt.writelines(str3)
#imgfile=GetFileList('val/test_dog')
#for img in imgfile:
# str4=img+' '+'0'+'\n'
# test_txt.writelines(str4)
#test_txt.close()
print("成功⽣成⽂件列表")
然后就能⽣成了⾃⼰想要的.txt ⽂件，⽣成的结果如下：
