Fix single channel png e2p

pillow does not accept hxwx1 numpy arrays, added squeeze to make it hxw for single channel images
sunset1995 · Dec 15, 2024 · 5eabc97 · 5eabc97
1 parent 472943a
commit 5eabc97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convert360 b/convert360
@@ -52,4 +52,4 @@ else:
     raise NotImplementedError('Unknown convertion')
 
 # Output image
-Image.fromarray(out.astype(np.uint8)).save(args.o)
+Image.fromarray(out.astype(np.uint8).squeeze()).save(args.o)