import os
import cv2

# ================== 配置 ==================
IMAGE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/error"   # 修改为你的图片文件夹路径
TARGET_W = 1920
TARGET_H = 1080
PREFIX = "img"     # 重命名前缀
EXT = ".png"       # 统一保存格式（.jpg / .png）
START_INDEX = 1    # 起始编号
# =========================================

# 支持的图片后缀
IMG_EXTS = (".jpg", ".jpeg", ".png", ".bmp", ".webp")

# 读取并排序（保证顺序稳定）
image_files = sorted([
    f for f in os.listdir(IMAGE_DIR)
    if f.lower().endswith(IMG_EXTS)
])

print(f"[INFO] 共找到 {len(image_files)} 张图片")

for idx, filename in enumerate(image_files, start=START_INDEX):
    img_path = os.path.join(IMAGE_DIR, filename)

    img = cv2.imread(img_path)
    if img is None:
        print(f"[WARN] 读取失败，跳过: {filename}")
        continue

    # resize
    resized = cv2.resize(img, (TARGET_W, TARGET_H), interpolation=cv2.INTER_AREA)

    # 新文件名
    new_name = f"{PREFIX}_{idx:04d}{EXT}"
    new_path = os.path.join(IMAGE_DIR, new_name)

    # 保存
    cv2.imwrite(new_path, resized)

    print(f"[OK] {filename} -> {new_name}")

print("[DONE] 全部处理完成")