From 221c80aa8c94a0fa389436dbbc94d6c6bbed5620 Mon Sep 17 00:00:00 2001 From: Martin Laasmaa Date: Thu, 11 Dec 2025 11:20:20 +0200 Subject: [PATCH] Small image showing fix --- src/gui/widgets/annotation_canvas_widget.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/gui/widgets/annotation_canvas_widget.py b/src/gui/widgets/annotation_canvas_widget.py index 880143e..e706a6f 100644 --- a/src/gui/widgets/annotation_canvas_widget.py +++ b/src/gui/widgets/annotation_canvas_widget.py @@ -246,10 +246,10 @@ class AnnotationCanvasWidget(QWidget): return try: - # Get RGB image data - if self.current_image.channels == 3: + # Get image data in a format compatible with Qt + if self.current_image.channels in (3, 4): image_data = self.current_image.get_rgb() - height, width, channels = image_data.shape + height, width = image_data.shape[:2] else: image_data = self.current_image.get_grayscale() height, width = image_data.shape