处理SUNRGBD数据

包含内容有　python的文件遍历、移动文件位置、修改文件名并重新排序

import os
import shutil

path = '/home/dell-g3/PycharmProjects/data/SUNRGBD/kv2/align_kv2/'

def get_filelist(dir, Filelist_depth, Filelist_image):
    if os.path.isdir(dir):
        for s in os.listdir(dir):
            dir = os.path.join(dir, s)
            for t in os.listdir(dir):
                # 如果需要忽略某些文件夹，使用以下代码
                if t == 'depth':
                    newDir_depth = os.path.join(dir, t)
                    tt = os.listdir(newDir_depth)
                    newDir_depth = os.path.join(newDir_depth, ''.join(tt))
                    Filelist_depth.append(newDir_depth)
                elif t == 'image':
                    newDir_image = os.path.join(dir, t)
                    tt = os.listdir(newDir_image)
                    newDir_image = os.path.join(newDir_image, ''.join(tt))
                    Filelist_image.append(newDir_image)
                else: continue
            dir = path
    return Filelist_depth, Filelist_image

if __name__ == '__main__':
    Filelist_depth, Filelist_image = get_filelist('/home/dell-g3/PycharmProjects/data/SUNRGBD/kv2/align_kv2/', [], [])
    print(len(Filelist_depth))
    j = 0
    for i in Filelist_depth:
        print(i)
        j = j+1
        newpath_depth = '/home/dell-g3/PycharmProjects/data/sun_rgbd/depth_train'
        shutil.move(i, newpath_depth)
        os.rename(i, newpath_depth + '/' + str(j) + '.png')
    print(len(Filelist_image))
    for i in Filelist_image:
        print(i)
        j = j+1
        newpath_image = '/home/dell-g3/PycharmProjects/data/sun_rgbd/image_train'
        shutil.move(i, newpath_image)
        os.rename(i, newpath_image + '/' + str(j) + '.jpg')