67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
|
import os
|
|||
|
|
import shutil
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
def copy_png_images(source_folder, destination_folder):
|
|||
|
|
"""
|
|||
|
|
从源文件夹递归搜索所有 .png 文件并复制到目标文件夹。
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
source_folder (str): 源文件夹路径
|
|||
|
|
destination_folder (str): 目标文件夹路径
|
|||
|
|
"""
|
|||
|
|
# 将字符串路径转换为 Path 对象，更方便操作
|
|||
|
|
src_path = Path(source_folder)
|
|||
|
|
dest_path = Path(destination_folder)
|
|||
|
|
|
|||
|
|
# 检查源文件夹是否存在
|
|||
|
|
if not src_path.exists():
|
|||
|
|
print(f"错误：源文件夹 '{source_folder}' 不存在。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if not src_path.is_dir():
|
|||
|
|
print(f"错误：'{source_folder}' 不是一个有效的文件夹。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 如果目标文件夹不存在，则创建它
|
|||
|
|
dest_path.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 用于统计复制的文件数量
|
|||
|
|
copied_count = 0
|
|||
|
|
|
|||
|
|
# 使用 rglob 递归搜索所有 .png 文件（不区分大小写）
|
|||
|
|
# rglob('*.[pP][nN][gG]') 可以匹配 .png, .PNG, .Png 等
|
|||
|
|
png_files = src_path.rglob('*.[jJ][pP][gG]')
|
|||
|
|
|
|||
|
|
for png_file in png_files:
|
|||
|
|
try:
|
|||
|
|
# 计算目标文件的完整路径
|
|||
|
|
# 保持源文件夹的相对结构，但只保留文件名（可选：如果想保持结构，去掉 .name）
|
|||
|
|
# 这里我们只复制文件名到目标文件夹，避免路径过长或结构复杂
|
|||
|
|
dest_file = dest_path / png_file.name
|
|||
|
|
|
|||
|
|
# 处理重名文件：在文件名后添加序号
|
|||
|
|
counter = 1
|
|||
|
|
original_dest_file = dest_file
|
|||
|
|
while dest_file.exists():
|
|||
|
|
dest_file = original_dest_file.parent / f"{original_dest_file.stem}_{counter}{original_dest_file.suffix}"
|
|||
|
|
counter += 1
|
|||
|
|
|
|||
|
|
# 执行复制
|
|||
|
|
shutil.copy2(png_file, dest_file) # copy2 会保留文件的元数据（如修改时间）
|
|||
|
|
print(f"已复制: {png_file} -> {dest_file}")
|
|||
|
|
copied_count += 1
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"复制文件时出错: {png_file} - {e}")
|
|||
|
|
|
|||
|
|
print(f"\n复制完成！共复制了 {copied_count} 个 PNG 文件到 '{destination_folder}'。")
|
|||
|
|
|
|||
|
|
# ------------------ 主程序 ------------------
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# ====== 请在这里修改源文件夹和目标文件夹的路径 ======
|
|||
|
|
source_folder = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/3" # 替换为你的源文件夹路径
|
|||
|
|
destination_folder = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/yolodataset/point" # 替换为你的目标文件夹路径
|
|||
|
|
# ====================================================
|
|||
|
|
|
|||
|
|
copy_png_images(source_folder, destination_folder)
|