mimc_rope / img.py
Qiyp's picture
add random cls ckpt
042a375
import pyarrow.parquet as pq
from PIL import Image
import io
table = pq.read_table("/home/v-ruoyufeng/v-ruoyufeng/qyp/datasets/downloads/ef4b5e6297f2e78af98b168d563b452438b711eb371489de4729f8e5d8168df8")
df = table.to_pandas()
from PIL import Image
import numpy as np
for idx, row in df.iterrows():
image_bytes = row['image_column'] # 假设图像数据在'image_column'列
image = Image.open(io.BytesIO(image_bytes))
image.save(f"kodim_{idx}.png")