check_label_list_widget.py

#!/usr/bin/env python

import sys

import imgviz
from qtpy import QtWidgets

from labelme.widgets import LabelListWidget
from labelme.widgets import LabelListWidgetItem


def main():
    app = QtWidgets.QApplication(sys.argv)

    view = LabelListWidget()

    colormap = imgviz.label_colormap(value=200)
    colormap = {
        "person": colormap[15],
        "dog": colormap[6],
    }
    for text in ["person", "dog", "dog", "dog", "person"]:
        item = LabelListWidgetItem()
        item.setText(
            '{} <font color="#{:02x}{:02x}{:02x}">●</font>'.format(
                text, *colormap[text]
            )
        )
        view.addItem(item)

    def itemChangedCallback(item):
        print("itemChangedCallback:", item, item.text())

    def itemSelectionChangedCallback(selected, deselected):
        for item in selected:
            print("itemSelectionChangedCallback:", item, item.text())

    def itemDoubleClickedCallback(item):
        print("itemDoubleClickedCallback:", item, item.text())

    view.itemDoubleClicked.connect(itemDoubleClickedCallback)
    view.itemChanged.connect(itemChangedCallback)
    view.itemSelectionChanged.connect(itemSelectionChangedCallback)

    view.show()
    view.raise_()
    sys.exit(app.exec_())


if __name__ == "__main__":
    main()