import pandas as pd
import os
def p_count(name):
path = 'D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/TrainingSet/trainingset/'+name+'/visible/'
img_list_v = sorted([p for p in os.listdir(path) if os.path.splitext(p)[1] in ['.jpg','.bmp','.png']])
img_list_v = [os.path.join(path) for img in img_list_v]
return len(img_list_v)
def save0(name,attribute,num):
#保存
#name:文件夹名字
#attribute:txt文件名
#num:打印数量
save_path = "D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/TrainingSet/trainingset/" + name+"/"+attribute+".txt"
txt = open(save_path,mode='w+')
i=0
while i < num:
txt.write('0\n')
i+=1
def save1(name,attribute,num):
save_path = "D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/TrainingSet/trainingset/" + name+"/"+attribute+".txt"
txt = open(save_path,mode='w+')
i=0
while i < num:
txt.write('1\n')
i+=1
def d0(name,attribute):
#删除路径
save_path = "D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/TrainingSet/trainingset/" + name + "/" + attribute + ".tag"
#删除命令
os.remove( save_path)
#读取
seq_path='D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/trainingsetList.txt'
f_seqlist=open(seq_path)
#遍历所有txt
for name in f_seqlist.readlines():
#name:图片文件夹名
name=name.strip('\n')
#读取每个文件属性
label_path = "D:/BaiduNetdiskDownload/LasHeR_Divided_TraningSet&TestingSet/AttriSeqsTxt/" + name+".txt"
label = pd.read_csv(label_path,dtype=object)
#print(label.columns[1])
#读取图片数量
n = p_count(name)
#确定属性
zhedang = "OCC"
guangzhaobinahua = "IV"
rejiaocha = "TC"
kuaisuyidong = "FM"
chidubiahua = "SV"
if float(label.columns[1])<1 and float(label.columns[2])<1:
# save0(name, zhedang, n)
d0(name,zhedang)
else:
# save1(name,zhedang,n)
# d0(name, zhedang)
pass
if float(label.columns[5])<1 and float(label.columns[6])<1:
# save0(name, guangzhaobinahua, n)
d0(name,guangzhaobinahua)
else:
# save1(name,guangzhaobinahua,n)
# d0(name, guangzhaobinahua)
pass
if float(label.columns[13])<1 :
# save0(name, rejiaocha, n)
d0(name,rejiaocha)
else:
# save1(name,rejiaocha,n)
# d0(name, rejiaocha)
pass
if float(label.columns[16])<1 :
# save0(name, kuaisuyidong, n)
d0(name,kuaisuyidong)
else:
# save1(name,kuaisuyidong,n)
# d0(name, kuaisuyidong)
pass
if float(label.columns[17])<1 :
# save0(name, chidubiahua, n)
d0(name,chidubiahua)
else:
# save1(name,chidubiahua,n)
# d0(name, chidubiahua)
pass
RGBT数据集制作
猜你喜欢
转载自blog.csdn.net/qq_45104603/article/details/127297995
今日推荐
周排行