import cv2
from pathlib import Path

# 路径设置
src_dir  = Path('dataset')          # 原始双目图文件夹
left_dir = src_dir / 'left'         # 输出左图
right_dir= src_dir / 'right'        # 输出右图
left_dir.mkdir(exist_ok=True)
right_dir.mkdir(exist_ok=True)

# 支持常见后缀
exts = ('*.jpg', '*.jpeg', '*.png', '*.bmp')
img_paths = [p for ext in exts for p in src_dir.glob(ext)]

if not img_paths:
    print('未找到任何图片，请检查 dataset/ 目录')
    exit()

for img_path in img_paths:
    img = cv2.imread(str(img_path))
    if img is None:
        print(f'读取失败：{img_path}')
        continue
    h, w = img.shape[:2]
    if w != 2560 or h != 720:
        print(f'跳过尺寸不符的图片：{img_path}  ({w}×{h})')
        continue

    left  = img[:, :1280]      # 左目
    right = img[:, 1280:]      # 右目

    # 保存
    cv2.imwrite(str(left_dir  / img_path.name), left)
    cv2.imwrite(str(right_dir / img_path.name), right)

print('全部处理完成！')