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)