UE渲染后整理成数据集

很简单的python文件操作，记录一下

import os
import shutil

# Define the source and destination directories
source_dir = os.path.join(os.path.dirname(__file__), '..', 'MovieRenders')
normal_dir = os.path.join(source_dir, 'test/normal')
color_dir = os.path.join(source_dir, 'test/color')
depth_dir = os.path.join(source_dir, 'test/depth')

# Create the destination directory if it doesn't exist
if not os.path.exists(normal_dir):
    os.makedirs(normal_dir)
if not os.path.exists(color_dir):
    os.makedirs(color_dir)
if not os.path.exists(depth_dir):
    os.makedirs(depth_dir)

# Iterate over all files in the source directory
for filename in os.listdir(source_dir):
    if 'WorldNormal' in filename:
        # Extract the numeric part and the file extension
        numeric_part = ''.join(filter(str.isdigit, filename))
        new_filename = numeric_part + '.jpg'
        
        # Move the file to the destination directory with the new name
        shutil.move(os.path.join(source_dir, filename), os.path.join(normal_dir, new_filename))
    elif 'WorldDepth' in filename:
        # delete the file
        os.remove(os.path.join(source_dir, filename))
    elif 'jpeg' in filename:
        # Extract the numeric part and the file extension
        numeric_part = ''.join(filter(str.isdigit, filename))
        new_filename = numeric_part + '.jpg'
        # Move the file to the destination directory with the new name
        shutil.move(os.path.join(source_dir, filename), os.path.join(color_dir, new_filename))
    elif 'exr' in filename:
        shutil.move(os.path.join(source_dir, filename), os.path.join(depth_dir, filename))