import os
from pathlib import Path

ROOT_PATH = Path("/teamspace/studios/this_studio/image-matching-webui")
prefix = "datasets/wxbs_benchmark/.WxBS/v1.1"
wxbs_path = ROOT_PATH / prefix

pairs = []
for catg in os.listdir(wxbs_path):
    catg_path = wxbs_path / catg
    if not catg_path.is_dir():
        continue
    for scene in os.listdir(catg_path):
        scene_path = catg_path / scene
        if not scene_path.is_dir():
            continue
        img1_path = scene_path / "01.png"
        img2_path = scene_path / "02.png"
        pairs.append([str(img1_path),str(img2_path)])

breakpoint()