70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
import os
|
||
import cv2
|
||
|
||
# =========================================================
|
||
# 配置
|
||
# =========================================================
|
||
|
||
SRC_DIR = "muban_image" # 原始模板目录
|
||
DST_DIR = "muban_image2" # 裁剪后保存目录
|
||
|
||
# 三个 ROI（x, y, w, h）
|
||
ROI_1 = (782, 614, 164, 128)
|
||
ROI_2 = (837, 791, 100, 99)
|
||
ROI_3 = (873, 736, 141, 110)
|
||
|
||
|
||
# =========================================================
|
||
# 裁剪函数
|
||
# =========================================================
|
||
def crop_and_save(img_path, save_dir):
|
||
img = cv2.imread(img_path)
|
||
if img is None:
|
||
print(f"[WARN] 读取失败: {img_path}")
|
||
return
|
||
|
||
h_img, w_img = img.shape[:2]
|
||
base = os.path.splitext(os.path.basename(img_path))[0]
|
||
|
||
roi_list = [ROI_1, ROI_2, ROI_3]
|
||
|
||
for idx, roi in enumerate(roi_list, start=1):
|
||
x, y, w, h = roi
|
||
|
||
# 边界保护
|
||
x1 = max(0, x)
|
||
y1 = max(0, y)
|
||
x2 = min(w_img, x + w)
|
||
y2 = min(h_img, y + h)
|
||
|
||
if x2 <= x1 or y2 <= y1:
|
||
print(f"[WARN] ROI_{idx} 超出图像范围: {img_path}")
|
||
continue
|
||
|
||
roi_img = img[y1:y2, x1:x2]
|
||
|
||
save_name = f"{base}_roi{idx}.png"
|
||
save_path = os.path.join(save_dir, save_name)
|
||
cv2.imwrite(save_path, roi_img)
|
||
|
||
print(f"[OK] 保存: {save_path}")
|
||
|
||
|
||
# =========================================================
|
||
# main
|
||
# =========================================================
|
||
if __name__ == "__main__":
|
||
if not os.path.isdir(SRC_DIR):
|
||
raise RuntimeError(f"源目录不存在: {SRC_DIR}")
|
||
|
||
os.makedirs(DST_DIR, exist_ok=True)
|
||
|
||
for fname in os.listdir(SRC_DIR):
|
||
if not fname.lower().endswith((".png", ".jpg", ".jpeg", ".bmp")):
|
||
continue
|
||
|
||
img_path = os.path.join(SRC_DIR, fname)
|
||
crop_and_save(img_path, DST_DIR)
|
||
|
||
print("\n[INFO] 所有图片裁剪完成")
|