from PIL import Image
import os
import json


image_dir = "/mnt/petrelfs/zhuchenglin/LLaVA/playground/data/LLaVA-Pretrain/images"
json_filename = "/mnt/petrelfs/zhuchenglin/LLaVA/playground/data/LLaVA-Pretrain/mscoco_gen_200k.json"

with open(json_filename, "r") as f:
    text_data = json.load(f)

for text in text_data:
    subdir = text["image"].split("/")[0]
    image_path = os.path.join(image_dir, text["image"])
    with Image.open(image_path) as img:
        print(f"Resolution of {image_path}: {img.size}")

print("所有图像已成功生成并保存。")