查找图像文件的小工具(查找图像文件的小工具在哪里)

实现的原由：平时我们在电脑上要查找一个文档，一般我们可以给出文件名、日期、甚至文件内容。但是如果说想要在电脑里找一个”猫“或”“狗”的图片，或者一个特别内容的图片，普通文档查找的方式就无法解决。

设计思路

1、在搜索图像前，把需要检测的图像，建立一个图像内容索引，存放在一个索引文件中。不用每次搜索都要智能检测。还有可以定期更新重建图像内容索引。

这里建立图像内容索引，使用的是百度大脑，百度AI开放平台，https://ai.baidu.com/，免费的API：图像识别接口。当然也可以自己训练自己的图像模型。本文只是提供一个实现想法思路。

2、建立好图像内容索引后，直接根据输入图像的关键词进行搜索。

界面UI

一个图像内容关键词输入框，一个文件列表框，一个label用来显示图像的，两个功能按钮，一个用来图像搜索，一个用来建立图像内容索引。

实现代码

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_())