主页 / 问题 / 1103899
Accepted
Shamus Rezol
Asked:2020-04-01 20:28:27 +0000 UTC2020-04-01 20:28:27 +0000 UTC

命名图像。蟒蛇3.7

  • 772

例如，有 250 多个小图像。它们被写在它们是什么上。例如，让我们随机抽取水果。我遇到了一个问题：我需要查看 250 张图像，用俄语写下它们的名称（手动；文本识别脚本不适合我的准确度），写下它们的文件路径并按名称对所有文件进行分组。最后，您应该得到以下 JSON 配置（非常粗略）：{Apple:["file_1.png", "file_185.png", ..], ..}。

正如您已经理解的那样，手动完成所有操作是不够的。

已经尝试过：

  1. 多线程：1个线程显示图像（cv2.imshow），在另一个我输入名称。没有成功。有很多不同的错误...

  2. 通过 cv2.waitKey 输入名称。没有成功。他非常不愿意认出 Unicode。不知何故，我尝试了一个“看似有效”的脚本，所以我发现它在文件中是胡说八道，而不是 Unicode 在 250 张图片之后......

  3. 通过 pygame 输入名称。Pygame 的 unicode 支持更差。我没有找到写 JEHHYUB、引号和我想要的破折号的方法。*

我有点绝望。什么可以帮助我实现目标？我重复。我需要西里尔文输入，我需要为此查看图像。

python

1 个回答

  1. Best Answer

    我扔了一个程序原型：

    • 控制台安装：pip install pyqt5
    • 有一个包含图片的文件夹：images
    • 通过启动时，glob会加载图片列表
    • 通过在键中指定值并单击添加图片，程序会将其添加到字典中，其中键将是输入的值，值将是图像文件的列表

    例子：

    import json
import glob

from PyQt5.QtWidgets import (
    QApplication, QMainWindow, QWidget, QDockWidget, QLabel, QLineEdit, QPushButton,
    QVBoxLayout, QHBoxLayout, QPlainTextEdit
)
from PyQt5.QtGui import QPixmap
from PyQt5.QtCore import Qt


class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self._current_index_image = 0
        self._images = glob.glob('images/*.png')
        self._data = dict()

        self.label_image = QLabel()
        self.label_image.setFixedSize(400, 400)
        self.label_image.setScaledContents(True)

        self.line_edit_key = QLineEdit()
        self.line_edit_key.setPlaceholderText('Введите ключ картинки...')

        self.button_add_file_name = QPushButton('Добавить картинку')
        self.button_add_file_name.clicked.connect(self._on_add_file_name)

        self.button_prev = QPushButton('Предыдущая картинка')
        self.button_prev.clicked.connect(self.load_prev_image)

        self.button_next = QPushButton('Следующая картинка')
        self.button_next.clicked.connect(self.load_next_image)

        layout_control = QHBoxLayout()
        layout_control.addWidget(QLabel('Ключ:'))
        layout_control.addWidget(self.line_edit_key)
        layout_control.addWidget(self.button_add_file_name)

        layout_control_2 = QHBoxLayout()
        layout_control_2.addWidget(self.button_prev)
        layout_control_2.addWidget(self.button_next)

        self.pl_text_json = QPlainTextEdit()

        self._dock_widget = QDockWidget('JSON')
        self._dock_widget.setWidget(self.pl_text_json)

        self.addDockWidget(Qt.RightDockWidgetArea, self._dock_widget)

        main_layout = QVBoxLayout()
        main_layout.addWidget(self.label_image)
        main_layout.addLayout(layout_control)
        main_layout.addLayout(layout_control_2)
        main_layout.addStretch()

        central_widget = QWidget()
        central_widget.setLayout(main_layout)

        self.setCentralWidget(central_widget)

        self.load_current_image()

    def get_current_image_file_name(self):
        return self._images[self._current_index_image]

    def load_prev_image(self):
        self._current_index_image -= 1
        if self._current_index_image < 0:
            self._current_index_image = 0

        self.load_current_image()

    def load_next_image(self):
        self._current_index_image += 1
        if self._current_index_image >= len(self._images):
            self._current_index_image = len(self._images) - 1

        self.load_current_image()

    def load_current_image(self):
        self.update_states()

        file_name = self.get_current_image_file_name()

        pixmap = QPixmap()
        pixmap.load(file_name)
        self.label_image.setPixmap(pixmap)

    def _on_add_file_name(self):
        key = self.line_edit_key.text()
        file_name = self.get_current_image_file_name()

        if key not in self._data:
            self._data[key] = []

        if file_name not in self._data[key]:
            self._data[key].append(file_name)

        self.pl_text_json.setPlainText(
            json.dumps(self._data, indent=4, ensure_ascii=False)
        )

    def update_states(self):
        file_name = self.get_current_image_file_name()
        self.setWindowTitle(f'{self._current_index_image + 1} / {len(self._images)} : {file_name}')


if __name__ == '__main__':
    app = QApplication([])

    mw = MainWindow()
    mw.show()

    app.exec()

    看起来像这样： 在此处输入图像描述

    • 1

相关问题