Bladeren bron

Handle when pixmap is not set yet

Kentaro Wada 1 jaar geleden
bovenliggende
commit
390d32c044
1 gewijzigde bestanden met toevoegingen van 4 en 0 verwijderingen
  1. 4 0
      labelme/widgets/canvas.py

+ 4 - 0
labelme/widgets/canvas.py

@@ -153,6 +153,10 @@ class Canvas(QtWidgets.QWidget):
                 ),
                 ),
             )
             )
 
 
+        if self.pixmap is None:
+            logger.warning("Pixmap is not set yet")
+            return
+
         self._ai_model.set_image(
         self._ai_model.set_image(
             image=labelme.utils.img_qt_to_arr(self.pixmap.toImage())
             image=labelme.utils.img_qt_to_arr(self.pixmap.toImage())
         )
         )