python 相似图片交互界面代码

以下是一个简单的基于 Python 的相似图片交互界面代码示例，使用了 PyQT5 和 Pillow 库：

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import QImage, QPixmap
from PIL import Image
import numpy as np


class SimilarImage(QWidget):
    def __init__(self):
        super().__init__()

        # 初始化UI
        self.initUI()

        # 图片相似度阈值
        self.threshold = 0.9

    def initUI(self):
        # 创建布局
        layout = QVBoxLayout()

        # 创建选择图片按钮
        self.btn_select = QPushButton('选择图片', self)
        self.btn_select.clicked.connect(self.select_image)
        layout.addWidget(self.btn_select)

        # 创建显示图片的标签
        self.label_image = QLabel(self)
        self.label_image.setFixedSize(400, 400)
        layout.addWidget(self.label_image)

        # 创建相似图片列表
        self.similar_images = QListWidget(self)
        self.similar_images.setFixedSize(400, 150)
        layout.addWidget(self.similar_images)

        # 创建相似度阈值输入框
        self.threshold_edit = QLineEdit(self)
        self.threshold_edit.setFixedSize(400, 30)
        self.threshold_edit.setText(str(self.threshold))
        self.threshold_edit.textChanged[str].connect(self.set_threshold)
        layout.addWidget(self.threshold_edit)

        # 设置布局
        self.setLayout(layout)

        # 设置窗口标题和大小
        self.setWindowTitle('相似图片检索')
        self.setFixedSize(400, 600)

    def select_image(self):
        # 打开文件对话框，选择图片文件
        options = QFileDialog.Options()
        options |= QFileDialog.DontUseNativeDialog
        file_name, _ = QFileDialog.getOpenFileName(self, '选择图片', '', 'Images (*.png *.xpm *.jpg *.bmp *.gif *.jpeg)', options=options)
        if file_name:
            # 加载图片
            img = Image.open(file_name).convert('RGB')

            # 将PIL图片转换为QImage
            qimage = QImage(img.tobytes(), img.size[0], img.size[1], QImage.Format_RGB888)

            # 将QImage显示在标签上
            pixmap = QPixmap(qimage)
            self.label_image.setPixmap(pixmap)

            # 计算相似图片
            self.find_similar_images(img)

    def find_similar_images(self, img):
        # TODO: 在这里实现相似图片检索算法

        # 添加相似图片到列表中
        self.similar_images.clear()
        self.similar_images.addItems(['similar_image1.jpg', 'similar_image2.jpg', 'similar_image3.jpg'])

    def set_threshold(self, text):
        try:
            self.threshold = float(text)
        except ValueError:
            pass


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = SimilarImage()
    ex.show()
    sys.exit(app.exec_())

请注意，这个示例代码中的 `find_similar_images` 方法仅展示了一个占位符，需要根据实际情况替换为相似图片检索算法的实现。