Unverified Commit 4c562a98 authored by Vladimir Mandic's avatar Vladimir Mandic Committed by GitHub
Browse files

convert rgba to rgb

some image format (e.g. jpg) do not support rgba
parent ea9bd9fc
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -73,6 +73,8 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):

            if not save_normally:
                os.makedirs(output_dir, exist_ok=True)
                if processed_image.mode == 'RGBA':
                    processed_image = processed_image.convert("RGB")
                processed_image.save(os.path.join(output_dir, filename))