Assign specific bg colors to widget item by different labels

2017-11-06 17:30:51 +08:00 · 2017-11-06 17:30:51 +08:00 · 821ffae6b9
commit 821ffae6b9
parent 7c145ef6e9
1 changed files with 1 additions and 0 deletions
--- a/labelImg.py
+++ b/labelImg.py
@ -664,6 +664,7 @@ class MainWindow(QMainWindow, WindowMixin):
        item = HashableQListWidgetItem(shape.label)
        item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
        item.setCheckState(Qt.Checked)
+        item.setBackground(generateColorByText(shape.label))
        self.itemsToShapes[item] = shape
        self.shapesToItems[shape] = item
        self.labelList.addItem(item)