helpers/download_tiles.py

from pathlib import Path
from datasets import load_dataset
from PIL import Image

def save_hf_tiles(dataset="Kratos-AI/KAI_car-images", split="train", out_dir="tiles", tile_size=32, limit=300):
    ds = load_dataset(dataset, split=split)
    out = Path(out_dir); out.mkdir(parents=True, exist_ok=True)
    n = 0
    for i, s in enumerate(ds):
        try:
            im = s["image"].convert("RGB").resize((tile_size, tile_size), Image.LANCZOS)
            im.save(out / f"hf_{i:05d}.jpg", quality=90)
            n += 1
            if limit and n >= limit:
                break
        except Exception:
            pass
    print(f"Saved {n} tiles → {out}/")

if __name__ == "__main__":
    save_hf_tiles()