From 3102f0a27d457148a4ccd7005204b227293c31f8 Mon Sep 17 00:00:00 2001 From: Markus Roth Date: Fri, 8 Nov 2013 23:53:39 +0100 Subject: [PATCH] Fix exception being raised when image is None. --- sloth/gui/labeltool.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/sloth/gui/labeltool.py b/sloth/gui/labeltool.py index 90caa31..dc65e80 100644 --- a/sloth/gui/labeltool.py +++ b/sloth/gui/labeltool.py @@ -130,6 +130,11 @@ class MainWindow(QMainWindow): img = self.labeltool.getImage(new_image) + if img == None: + self.controls.setFilename("") + self.selectionmodel.setCurrentIndex(new_image.index(), QItemSelectionModel.ClearAndSelect|QItemSelectionModel.Rows) + return + h = img.shape[0] w = img.shape[1] self.image_resolution.setText("%dx%d" % (w, h))