- 添加了类别查找功能。

  当类别很多时，可通过该功能快速找到指定类别。查找使用字符串模糊匹配。

Author: yatengLG

ISAT/ui/category_dock.py

@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
 
-# Form implementation generated from reading ui file 'category_dock.ui'
+# Form implementation generated from reading ui file '/home/lg/PycharmProjects/ISAT_with_segment_anything/ISAT/ui/category_dock.ui'
 #
-# Created by: PyQt5 UI code generator 5.15.9
+# Created by: PyQt5 UI code generator 5.15.10
 #
 # WARNING: Any manual changes made to this file will be lost when pyuic5 is
 # run again.  Do not edit this file unless you know what you are doing.
@@ -17,8 +17,11 @@ def setupUi(self, Form):
         Form.resize(294, 462)
         self.verticalLayout = QtWidgets.QVBoxLayout(Form)
         self.verticalLayout.setContentsMargins(2, 2, 2, 2)
-        self.verticalLayout.setSpacing(0)
+        self.verticalLayout.setSpacing(6)
         self.verticalLayout.setObjectName("verticalLayout")
+        self.lineEdit_search_category = QtWidgets.QLineEdit(Form)
+        self.lineEdit_search_category.setObjectName("lineEdit_search_category")
+        self.verticalLayout.addWidget(self.lineEdit_search_category)
         self.listWidget = QtWidgets.QListWidget(Form)
         self.listWidget.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)
         self.listWidget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
@@ -55,6 +58,7 @@ def setupUi(self, Form):
     def retranslateUi(self, Form):
         _translate = QtCore.QCoreApplication.translate
         Form.setWindowTitle(_translate("Form", "Form"))
+        self.lineEdit_search_category.setPlaceholderText(_translate("Form", "Search categories."))
         self.pushButton_group_mode.setStatusTip(_translate("Form", "Group mode"))
         self.pushButton_group_mode.setWhatsThis(_translate("Form", "Group mode"))
         self.pushButton_group_mode.setText(_translate("Form", "Auto"))

ISAT/ui/category_dock.ui

@@ -15,7 +15,7 @@
   </property>
   <layout class="QVBoxLayout" name="verticalLayout">
    <property name="spacing">
-    <number>0</number>
+    <number>6</number>
    </property>
    <property name="leftMargin">
     <number>2</number>
@@ -29,6 +29,13 @@
    <property name="bottomMargin">
     <number>2</number>
    </property>
+   <item>
+    <widget class="QLineEdit" name="lineEdit_search_category">
+     <property name="placeholderText">
+      <string>Search categories.</string>
+     </property>
+    </widget>
+   </item>
    <item>
     <widget class="QListWidget" name="listWidget">
      <property name="selectionMode">

ISAT/ui/zh_CN.qm

@@ -927,37 +927,37 @@
         <translation>显示/隐藏</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="59"/>
+        <location filename="category_dock.py" line="63"/>
         <source>Group mode</source>
         <translation>组模式[自动/手动]</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="60"/>
+        <location filename="category_dock.py" line="64"/>
         <source>Auto</source>
         <translation></translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="62"/>
+        <location filename="category_dock.py" line="66"/>
         <source>Current group -</source>
         <translation>组id-</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="63"/>
+        <location filename="category_dock.py" line="67"/>
         <source>-</source>
         <translation>-</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="65"/>
+        <location filename="category_dock.py" line="69"/>
         <source>Current group</source>
         <translation>当前组id</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="67"/>
+        <location filename="category_dock.py" line="71"/>
         <source>Current group +</source>
         <translation>组id+</translation>
     </message>
     <message>
-        <location filename="category_dock.py" line="68"/>
+        <location filename="category_dock.py" line="72"/>
         <source>+</source>
         <translation>+</translation>
     </message>
@@ -1006,6 +1006,11 @@
         <source>Next group.</source>
         <translation>下一组.</translation>
     </message>
+    <message>
+        <location filename="category_dock.py" line="61"/>
+        <source>Search categories.</source>
+        <translation>查找类别.</translation>
+    </message>
 </context>
 <context>
     <name>MainWindow</name>

ISAT/ui/zh_CN.ts

@@ -3,6 +3,7 @@
 
 from PyQt5 import QtWidgets, QtCore
 from ISAT.ui.category_dock import Ui_Form
+from fuzzywuzzy import process
 
 
 class CategoriesDockWidget(QtWidgets.QWidget, Ui_Form):
@@ -11,6 +12,8 @@ def __init__(self, mainwindow):
         self.setupUi(self)
         self.mainwindow = mainwindow
         self.listWidget.itemClicked.connect(self.item_choice)
+        self.lineEdit_search_category.textChanged.connect(self.update_widget)
+        self.lineEdit_search_category.setClearButtonEnabled(True)
 
         # 新增 手动/自动 group 选择
         self.lineEdit_currentGroup.setText(str(self.mainwindow.current_group))
@@ -23,8 +26,21 @@ def update_widget(self):
         self.listWidget.clear()
         btngroup = QtWidgets.QButtonGroup(self)
         labels = self.mainwindow.cfg.get('label', [])
-        for index in range(len(labels)):
-            label = labels[index]
+        search_text = self.lineEdit_search_category.text()
+
+        name_label_dict = {label.get('name', 'UNKNOW'): label for label in labels}
+
+        label_names = [label.get('name', 'UNKNOW') for label in labels]
+        if search_text == '':
+            show_label_names = label_names
+        elif search_text.strip(' ') == '':
+            show_label_names = label_names
+        else:
+            matches = process.extract(search_text, label_names, limit=5)
+            show_label_names = [name for name, score in matches if score > 0]
+
+        for index in range(len(show_label_names)):
+            label = name_label_dict[show_label_names[index]]
             name = label.get('name', 'UNKNOW')
             color = label.get('color', '#000000')
             item = QtWidgets.QListWidgetItem()

ISAT/widgets/category_dock_widget.py

@@ -11,4 +11,6 @@ numpy
 pycocotools
 shapely
 hydra-core>=1.3.2
-tqdm>=4.66.1
+tqdm>=4.66.1
+fuzzywuzzy
+python-Levenshtein

requirements.txt

@@ -59,7 +59,9 @@ def get_install_requires():
         'timm',
         'shapely',
         'hydra-core>=1.3.2',
-        'tqdm>=4.66.1'
+        'tqdm>=4.66.1',
+        'fuzzywuzzy'
+        'python-Levenshtein'
         ],
 
     classifiers=[