Merge pull request #20 from johnnylu305/main

Save as png format instead of jpg format
kakaobrain · Jan 29, 2023 · 31d7469 · 31d7469
2 parents ac10296 + 326bc7f
commit 31d7469
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/store_image.py b/utils/store_image.py
@@ -22,7 +22,7 @@ def norm8b(x):
 
 def store_image(dirpath, rgbs):
  for (i, rgb) in enumerate(rgbs):
- imgname = f"image{str(i).zfill(3)}.jpg"
+ imgname = f"image{str(i).zfill(3)}.png"
  rgbimg = Image.fromarray(to8b(rgb.detach().cpu().numpy()))
  imgpath = os.path.join(dirpath, imgname)
  rgbimg.save(imgpath)