Update check state of the annotated image in saveLabels

tinyloop · Jul 30, 2018 · 90ba63d · 90ba63d
1 parent 5b3a70f
commit 90ba63d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/labelme/app.py b/labelme/app.py
@@ -787,6 +787,10 @@ def format_shape(s):
                 flags=flags,
             )
             self.labelFile = lf
+            item = self.fileListWidget.findItems(
+                self.imagePath, Qt.MatchExactly
+            )[0]
+            item.setCheckState(Qt.Checked)
             # disable allows next and previous image to proceed
             # self.filename = filename
             return True