martin/object-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small image showing fix

Browse Source
This commit is contained in:
Martin Laasmaa
2025-12-11 11:20:20 +02:00
parent 833b222fad
commit 221c80aa8c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/gui/widgets/annotation_canvas_widget.py
View File

@@ -246,10 +246,10 @@ class AnnotationCanvasWidget(QWidget):
return return
try: try:
# Get RGB image data # Get image data in a format compatible with Qt
if self.current_image.channels == 3: if self.current_image.channels in (3, 4):
image_data = self.current_image.get_rgb() image_data = self.current_image.get_rgb()
height, width, channels = image_data.shape height, width = image_data.shape[:2]
else: else:
image_data = self.current_image.get_grayscale() image_data = self.current_image.get_grayscale()
height, width = image_data.shape height, width = image_data.shape