python 中显示图像的方法（72.Python基于PyQt）
python 中显示图像的方法（72.Python基于PyQt）查看图像建立图像索引一个图像内容关键词输入框，一个文件列表框，一个label用来显示图像的，两个功能按钮，一个用来图像搜索，一个用来建立图像内容索引。UIimport sys import os from PyQt5 import QtCore QtGui QtWidgets from Ui_findimage import Ui_MainWindow from aip import AipImageClassify import cv2 import json class MainWindow(QtWidgets.QMainWindow Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) self.btnfind.clicked.con
实现的原由：平时我们在电脑上要查找一个文档，一般我们可以给出文件名、日期、甚至文件内容。但是如果说想要在电脑里找一个”猫“或”“狗”的图片，或者一个特别内容的图片，普通文档查找的方式就无法解决。
设计思路1、在搜索图像前，把需要检测的图像，建立一个图像内容索引，存放在一个索引文件中。不用每次搜索都要智能检测。还有可以定期更新重建图像内容索引。
这里建立图像内容索引，使用的是百度大脑，百度AI开放平台，https://ai.baidu.com/，免费的API：图像识别接口。当然也可以自己训练自己的图像模型。本文只是提供一个实现想法思路。
2、建立好图像内容索引后，直接根据输入图像的关键词进行搜索。
界面UI一个图像内容关键词输入框，一个文件列表框，一个label用来显示图像的，两个功能按钮，一个用来图像搜索，一个用来建立图像内容索引。
UI
实现代码import sys
import os
from PyQt5 import QtCore QtGui QtWidgets
from Ui_findimage import Ui_MainWindow
from aip import AipImageClassify
import cv2
import json
class MainWindow(QtWidgets.QMainWindow Ui_MainWindow):
def __init__(self):
super().__init__()
self.setupUi(self)
self.btnfind.clicked.connect(self.findimg)
self.btnindex.clicked.connect(self.reindex)
self.reslist=[]
#建立图像内容索引
def reindex(self):
lstimg=[] #图像文件列表
lstidx=[] #图像内容索引列表
#选择文件夹
dirname=QtWidgets.QFileDialog.getExistingDirectory(self '选择文件夹')
#获取文件夹中的所有图像文件
for root dirs files in os.walk(dirname):
for file in files:
if os.path.splitext(file)[1] in ['.jpg' '.png' '.bmp']:
lstimg.append(os.path.join(root file))
self.statusBar().showMessage('正在建立图像内容索引...' file)
keyindex=self.recobj(os.path.join(root file))
if keyindex:
lstidx.append(keyindex)
QtWidgets.QApplication.processEvents()
#把结果放到listview中
slm=QtCore.QStringListModel()
slm.setStringList(lstimg)
self.filelistview.setModel(slm)
#默认显示第一张图片
if len(lstimg)>0:
#设置filelistview的当前索引为第一个
self.filelistview.setCurrentIndex(slm.index(0))
self.on_filelistview_clicked(self.filelistview.currentIndex())
#把列表写入文本文件 每行一个图像文件名 编码为utf-8
with open('index.txt' 'w' encoding="utf-8") as f:
for i in lstidx:
#字典转换为json字符串 编码为utf-8
sk=json.dumps(i ensure_ascii=False)
f.write(sk '\n')
self.statusbar.showMessage('图像内容索引已建立')
#图片识别
def recobj(self imgpath):
""" 这里输入你创建应用获得的三个参数"""，#请自行申请
APP_ID = '**********'
API_KEY = '******************'
SECRET_KEY = '***************'
client = AipImageClassify(APP_ID API_KEY SECRET_KEY)
#plantDetect carDetect logoSearch animalDetect dishDetect advancedGeneral
""" 调用通用物体识别 """
img=cv2.imread(imgpath)
gray=cv2.cvtColor(img cv2.COLOR_BGR2GRAY)
fimg=cv2.imencode('.jpg' gray)[1].tobytes()
try:
result=client.advancedGeneral(fimg)
#print(result)
if len(result['result'])>0:
for res in result['result']:
if res['score']>0.5:
dkey={imgpath:res['keyword']}
self.reslist.append(dkey)
except Exception as e:
print(e)
return
return self.reslist
def findimg(self):
lst=[] #结果图像文件列表
keyword=self.editkeyword.text().strip()
#判断图像内容索引是否存在
if not os.path.exists('index.txt'):
#提示图像内容索引不存在 请先建立图像内容索引 弹出对话框
reply=QtWidgets.QMessageBox.question(self '提示' '图像内容索引不存在 请先建立图像内容索引' QtWidgets.QMessageBox.Yes QtWidgets.QMessageBox.No)
else:
#打开图像内容索引文件
with open('index.txt' 'r' encoding="utf-8") as f:
line=f.readline()
for ditem in eval(line):
#获取字典中的键值和键值对应的值
for k v in ditem.items():
print(k v)
if keyword in v:
if k not in lst:
lst.append(k)
#把结果放到listview中
slm=QtCore.QStringListModel()
slm.setStringList(lst)
self.filelistview.setModel(slm)
#默认显示第一张图片
if len(lst)>0:
#设置filelistview的当前索引为第一个
self.filelistview.setCurrentIndex(slm.index(0))
self.on_filelistview_clicked(self.filelistview.currentIndex())
#设置listview的单击事件
def on_filelistview_clicked(self index):
#获取选中的图像文件的路径
filename=self.filelistview.model().stringList()[index.row()]
#设置标签的宽度
self.lblimg.setFixedWidth(self.filelistview.width())
#设置标签的图像自适应
#self.lblimg.setScaledContents(True)
#设置标签的图像按比例缩放
self.lblimg.setPixmap(QtGui.QPixmap(filename).scaled(self.lblimg.width() self.lblimg.height() QtCore.Qt.KeepAspectRatio))
#设置图像文件的标题
self.statusbar.showMessage(filename)
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
window = MainWindow()
window.setWindowTitle('FindImage')
window.show()
sys.exit(app.exec_())运行测试
未建图像内容索引提示
建立图像索引
查看图像
搜索：大象
搜索：车
程序已打包成单独的可执行文件，有需要的可以留言或私信。
