1. 程式人生 > >讀取資料夾下非隱藏檔案及SBU資料rgb提取

讀取資料夾下非隱藏檔案及SBU資料rgb提取

函式 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)