import os
from PIL import Image
from tqdm import tqdm

def align_to_16(path):
    print(f"🛠️  正在对 {path} 进行物理对齐（居中裁剪至16倍数）...")
    if not os.path.exists(path):
        print("❌ 目录不存在！")
        return
        
    files = sorted([f for f in os.listdir(path) if f.lower().endswith(('.png', '.jpg', '.jpeg'))])
    for f in tqdm(files):
        img_path = os.path.join(path, f)
        with Image.open(img_path) as img:
            w, h = img.size
            # 计算最大的 16 倍数
            new_w = (w // 16) * 16
            new_h = (h // 16) * 16
            
            if w != new_w or h != new_h:
                # 居中裁剪逻辑
                left = (w - new_w) // 2
                top = (h - new_h) // 2
                right = left + new_w
                bottom = top + new_h
                img.crop((left, top, right, bottom)).save(img_path)

# 执行物理修复
align_to_16("/root/autodl-tmp/dataset/deepblending_clean/DrJohnson/images")