RGBT数据集制作

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