读取文件夹下非隐藏文件及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)