diff --git a/labelme/app.py b/labelme/app.py index d73e2857bb95a916086193a50eeda8ad28510bd2..452d9a352e094be9b218b9acc2df5127adff9b5a 100644 --- a/labelme/app.py +++ b/labelme/app.py @@ -1,3 +1,5 @@ +# -*- coding: utf-8 -*- + import functools import os import os.path as osp @@ -963,11 +965,16 @@ class MainWindow(QtWidgets.QMainWindow): text = shape.label else: text = '{} ({})'.format(shape.label, shape.group_id) - item = QtWidgets.QListWidgetItem(text) + item = QtWidgets.QListWidgetItem() item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(Qt.Checked) self.labelList.itemsToShapes.append((item, shape)) self.labelList.addItem(item) + qlabel = QtWidgets.QLabel() + qlabel.setText(text) + qlabel.setAlignment(QtCore.Qt.AlignBottom) + item.setSizeHint(qlabel.sizeHint()) + self.labelList.setItemWidget(item, qlabel) if not self.uniqLabelList.findItems(shape.label, Qt.MatchExactly): self.uniqLabelList.addItem(shape.label) self.labelDialog.addLabelHistory(shape.label) @@ -984,6 +991,10 @@ class MainWindow(QtWidgets.QMainWindow): r, g, b = self._config['default_shape_color'] else: return + qlabel.setText( + '{} ' + .format(text, r, g, b) + ) shape.line_color = QtGui.QColor(r, g, b) shape.vertex_fill_color = QtGui.QColor(r, g, b) shape.hvertex_fill_color = QtGui.QColor(255, 255, 255)