42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
import os
|
|||
|
|
import shutil
|
|||
|
|
|
|||
|
|
def overwrite_images(folder1, folder2):
|
|||
|
|
"""
|
|||
|
|
在 folder1 和 folder2 中查找同名图片，
|
|||
|
|
用 folder1 的图片覆盖 folder2 中的同名图片。
|
|||
|
|
"""
|
|||
|
|
# 确保两个路径存在
|
|||
|
|
if not os.path.exists(folder1) or not os.path.exists(folder2):
|
|||
|
|
print("❌ 输入的文件夹路径不存在")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 获取 folder1 所有文件
|
|||
|
|
files1 = set(os.listdir(folder1))
|
|||
|
|
files2 = set(os.listdir(folder2))
|
|||
|
|
|
|||
|
|
# 找出交集（相同名字的文件）
|
|||
|
|
common_files = files1 & files2
|
|||
|
|
|
|||
|
|
if not common_files:
|
|||
|
|
print("⚠️ 没有找到同名文件")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
for file in common_files:
|
|||
|
|
src = os.path.join(folder1, file)
|
|||
|
|
dst = os.path.join(folder2, file)
|
|||
|
|
|
|||
|
|
# 仅处理图片文件（可根据需要扩展）
|
|||
|
|
if src.lower().endswith((".jpg", ".jpeg", ".png", ".bmp", ".tif", ".tiff")):
|
|||
|
|
shutil.copy2(src, dst)
|
|||
|
|
print(f"✅ 已覆盖: {dst}")
|
|||
|
|
|
|||
|
|
print("🎯 覆盖完成！")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
folder1 = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/classdata1/save/class4" # 源文件夹
|
|||
|
|
folder2 = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/seg/resize_seg2/train" # 目标文件夹
|
|||
|
|
|
|||
|
|
overwrite_images(folder1, folder2)
|