first commit

ailai_pc/cover_image.py

@@ -0,0 +1,41 @@
+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)