Browse Source

Fix delete item

Kentaro Wada 5 years ago
parent
commit
c0d1b2616a
2 changed files with 5 additions and 1 deletions
  1. 1 1
      labelme/app.py
  2. 4 0
      labelme/widgets/label_list_widget.py

+ 1 - 1
labelme/app.py

@@ -1009,7 +1009,7 @@ class MainWindow(QtWidgets.QMainWindow):
     def remLabels(self, shapes):
         for shape in shapes:
             item = self.labelList.findItemByShape(shape)
-            self.labelList.takeItem(self.labelList.row(item))
+            self.labelList.removeItem(item)
 
     def loadShapes(self, shapes, replace=True):
         self._noSelectionSlot = True

+ 4 - 0
labelme/widgets/label_list_widget.py

@@ -157,6 +157,10 @@ class LabelListWidget(QtWidgets.QListView):
         self.model().setItem(self.model().rowCount(), 0, item)
         item.setSizeHint(self.itemDelegate().sizeHint(None, None))
 
+    def removeItem(self, item):
+        index = self.model().indexFromItem(item)
+        self.model().removeRows(index.row(), 1)
+
     def selectItem(self, item):
         index = self.model().indexFromItem(item)
         self.selectionModel().select(index, QtCore.QItemSelectionModel.Select)