45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
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] 全部处理完成")
|