pytorch的数据集的相关代码(dataloaders)

1.遍历文件夹得到文件的列表

 

import glob
import os
gt = glob.glob("./gt/*.png") #遍历该文件夹下的所有的文件，得到一个列表
gt_name = [_[5:-7] for _ in gt]
gt_name_set = set(gt_name)#将列表转为集合

for _ in gt_name_set:
    command = 'ls ./gt/'+_+'* | grep -c .png'
    if os.popen(command).read().find('100')!=-1:#os.popen执行命令并得到返回值
        command_makedir = 'mkdir ./gt/'+_
        os.system(command_makedir)#os.system执行命令，不去获取返回值
        command_mvpngs = 'mv ./gt/'+_+'*.png ./gt/'+_
        os.system(command_mvpngs)