Show if a file is annotated by check box

zhangjun135 · Jul 20, 2018 · 366dd48 · 366dd48
1 parent 4d64f02
commit 366dd48
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/labelme/app.py b/labelme/app.py
@@ -1317,8 +1317,15 @@ def importDirImages(self, dirpath, load=True):
         self.lastOpenDir = dirpath
         self.filename = None
         self.fileListWidget.clear()
-        for imgPath in self.scanAllImages(dirpath):
-            item = QtWidgets.QListWidgetItem(imgPath)
+        for filename in self.scanAllImages(dirpath):
+            label_file = os.path.splitext(filename)[0] + '.json'
+            item = QtWidgets.QListWidgetItem(filename)
+            item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
+            if QtCore.QFile.exists(label_file) and \
+                    LabelFile.isLabelFile(label_file):
+                item.setCheckState(Qt.Checked)
+            else:
+                item.setCheckState(Qt.Unchecked)
             self.fileListWidget.addItem(item)
         self.openNextImg(load=load)