python copy picture

import os
import shutil

# 主目录路径
main_dir = '/data/Dgh/lung/data'
# 目标目录路径
target_dir = '/data/Dgh/lung/Paddle-CheXNet-main/dataset/images'
# 需要复制的图片扩展名列表
extensions = ['.jpg', '.png', '.gif', '.bmp']

# 创建目标目录
if not os.path.exists(target_dir):
    os.makedirs(target_dir)

# 复制图片到目标目录
for subdir, dirs, files in os.walk(main_dir):
    for file in files:
        print("file:", file)
        if os.path.splitext(file)[1].lower() in extensions:
            source_file_path = os.path.join(subdir, file)
            print("source_file_path:", source_file_path)
            # 目标文件的路径，确保不会覆盖同名文件
            target_file_path = os.path.join(target_dir, file)
            print("target_file_path:", target_file_path)
            # 避免文件名冲突
            if os.path.exists(target_file_path):
                # 如果需要处理文件名冲突，可以在这里添加逻辑，例如添加序列号
                pass
            else:
                shutil.copy2(source_file_path, target_dir)

print("图片复制完成。")