oops / convert.py
deansmile123's picture
Upload folder using huggingface_hub
75f0bc0 verified
Raw
History Blame Contribute Delete
671 Bytes
import os
from PIL import Image
from pillow_heif import register_heif_opener
register_heif_opener()
src = "/vast/ds5725/OOPS/OOPS_raw_data"
dst = "/vast/ds5725/OOPS/images"
os.makedirs(dst, exist_ok=True)
for fname in os.listdir(src):
if fname.lower().endswith(".jpg") or fname.lower().endswith(".jpeg"):
jpg_path = os.path.join(src, fname)
png_name = os.path.splitext(fname)[0] + ".png"
png_path = os.path.join(dst, png_name)
if os.path.exists(png_path):
print(f"Skipping (exists): {png_path}")
continue
img = Image.open(jpg_path).convert("RGB")
img.save(png_path, "PNG")
print("Done!")