Ignore the fourth channel/mask when saving an image

The renderer returns a RGBA image, just save the first 3 channels to correctly save the image of the resulting edits.
heycms · Jun 26, 2023 · abd757f · abd757f
1 parent d7f7319
commit abd757f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/viz/capture_widget.py b/viz/capture_widget.py
@@ -31,7 +31,6 @@ def dump_png(self, image):
         viz = self.viz
         try:
             _height, _width, channels = image.shape
-            assert channels in [1, 3]
             assert image.dtype == np.uint8
             os.makedirs(self.path, exist_ok=True)
             file_id = 0
@@ -43,7 +42,7 @@ def dump_png(self, image):
             if channels == 1:
                 pil_image = PIL.Image.fromarray(image[:, :, 0], 'L')
             else:
-                pil_image = PIL.Image.fromarray(image, 'RGB')
+                pil_image = PIL.Image.fromarray(image[:, :, :3], 'RGB')
             pil_image.save(os.path.join(self.path, f'{file_id:05d}.png'))
         except:
             viz.result.error = renderer.CapturedException()