最新推送

zhuangtai_class_cls/resize—renyi.py

@@ -0,0 +1,74 @@
+import os
+import cv2
+
+# ----------------------------
+# 配置
+# ----------------------------
+SOURCE_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/cls-61/class4"
+TARGET_ROOT_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/cls-61/class4c"
+TARGET_SIZE = 640  # resize 尺寸
+
+# ----------------------------
+# 全局 ROI (x, y, w, h)
+# ----------------------------
+GLOBAL_ROI = [445, 540, 931, 319]
+#GLOBAL_ROI = [604, 182, 594, 252]
+
+
+# ----------------------------
+# 主处理函数
+# ----------------------------
+def process_images():
+    x, y, w, h = GLOBAL_ROI
+
+    # 遍历主目录下的所有子文件夹
+    for sub_dir_name in sorted(os.listdir(SOURCE_ROOT_DIR)):
+        src_sub_dir = os.path.join(SOURCE_ROOT_DIR, sub_dir_name)
+
+        if not os.path.isdir(src_sub_dir):
+            continue  # 跳过非文件夹
+
+        tgt_sub_dir = os.path.join(TARGET_ROOT_DIR, sub_dir_name)
+        os.makedirs(tgt_sub_dir, exist_ok=True)
+
+        print(f"\n📁 处理文件夹: {sub_dir_name}")
+
+        for file in os.listdir(src_sub_dir):
+            if not file.lower().endswith((".jpg", ".png", ".jpeg", ".bmp", ".tif", ".tiff")):
+                continue
+
+            img_path = os.path.join(src_sub_dir, file)
+            img = cv2.imread(img_path)
+
+            if img is None:
+                print(f"❌ 无法读取图片: {img_path}")
+                continue
+
+            h_img, w_img = img.shape[:2]
+
+            # ROI 边界保护
+            x1 = max(0, x)
+            y1 = max(0, y)
+            x2 = min(w_img, x + w)
+            y2 = min(h_img, y + h)
+
+            cropped = img[y1:y2, x1:x2]
+            if cropped.size == 0:
+                print(f"❌ 裁剪结果为空: {img_path}")
+                continue
+
+            resized = cv2.resize(cropped, (TARGET_SIZE, TARGET_SIZE), interpolation=cv2.INTER_AREA)
+
+            tgt_path = os.path.join(tgt_sub_dir, file)
+            cv2.imwrite(tgt_path, resized)
+
+            print(f"✅ {sub_dir_name}/{file}")
+
+    print("\n🎉 全部图片处理完成")
+
+
+# ----------------------------
+# 入口
+# ----------------------------
+if __name__ == "__main__":
+    process_images()