讀取資料夾下非隱藏檔案及SBU資料rgb提取
阿新 • • 發佈:2018-12-17
函式 listdir_nohidden讀取檔案時去掉隱藏檔案
import os import shutil def listdir_nohidden(path): p = [] for f in os.listdir(path): if not f.startswith('.'): p.append(f) return p rootdir = '/home/jianyuchen/pyproject/data/SBU' sbu = listdir_nohidden(rootdir) sbu.sort() #/home/jianyuchen/pyproject/data/SBU/s01s02/s01s02/01/001 train_dir = '/home/jianyuchen/pyproject/pytorch-video-recognition-master/dataloaders/SBU/train' for Tpeo in sbu[0:17]: #print(i) dir0 = rootdir + '/' + Tpeo +'/'+ Tpeo classz = listdir_nohidden(dir0) classz.sort() for clas in classz: dir1 = dir0 +'/'+ clas print(dir1) group = listdir_nohidden(dir1) group.sort() print(group) for grou in group: dir2 = dir1 + '/' + grou print(dir2) jpgs = os.listdir(dir2) jpgs.sort() for jpg in jpgs: if (jpg.split('_')[0]=='rgb'): traindir=os.path.join(train_dir, clas,Tpeo+grou) if os.path.exists(traindir)==False: os.mkdir(traindir) shutil.copy(os.path.join(dir2,jpg), traindir)