Fixing pseudo rgb

src/gui/widgets/annotation_canvas_widget.py

@@ -250,12 +250,10 @@ class AnnotationCanvasWidget(QWidget):
             # 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 = image_data.shape[:2]
             else:
-                image_data = self.current_image.get_grayscale()
-                height, width = image_data.shape
+                image_data = self.current_image.get_qt_rgb()
 
-            image_data = np.ascontiguousarray(image_data)
+            height, width = image_data.shape[:2]
             bytes_per_line = image_data.strides[0]
 
             qimage = QImage(
@@ -263,7 +261,7 @@ class AnnotationCanvasWidget(QWidget):
                 width,
                 height,
                 bytes_per_line,
-                self.current_image.qtimage_format,
+                QImage.Format_RGBX32FPx4,  # self.current_image.qtimage_format,
             ).copy()  # Copy so Qt owns the buffer even after numpy array goes out of scope
 
             self.original_pixmap = QPixmap.fromImage(qimage)