对图片进行索引，存入数据库sqlite3中，实现快速搜索打开

对图片进行索引，存入数据库中，实现快速搜索打开

  

这个任务分为两步：

第一步：建立索引

 import os
 import shutil
 import sqlite3
 # 扫描函数,需扫描路径目录处理该目录及下级目录所有图片地址,简单明了
 
 def scanDirSearchFile(dir, dbname):
     connsqlite3 = sqlite3.connect(dbname)
     cu = connsqlite3.cursor()
     for root, dirs, files in os.walk(dir, True, None, False):  # 遍列目录
         # 处理该文件夹下所有文件:
 
         for f in files:
             if os.path.isfile(os.path.join(root, f)):
                 ext = os.path.splitext(f)[1].lower()
                 if ext in ('.jpg', '.png', '.bmp'):
                     # print os.path.join(root,f)
                     cu.execute("insert into photoaddress(photoaddress,photoname) values(?,?);",
                                ((os.path.join(root, f)), (f)))
                     pass
     connsqlite3.commit()
 # 初始化数据库表
 def CreatePhotoDB(sqliteName):
    # 数据库文件处理
     connsqlite3 = sqlite3.connect(sqliteName)
     cu = connsqlite3.cursor()
     # 创建表
     sql = "create table IF NOT EXISTS photoaddress(photoaddress varchar(400),photoname varchar(200));"
     cu.execute(sql)
     # 创建图片地址索引
     sql = "create index IF NOT EXISTS photoaddress_photoaddress  on  photoaddress(photoaddress);"
     cu.execute(sql)
     # 创建图片名索引
     sql = "create index IF NOT EXISTS photoaddress_photoname  on  photoaddress(photoname);"
     cu.execute(sql)
     # 清空库表
     sql = "delete from photoaddress;"
     cu.execute(sql)
     connsqlite3.commit()
 if __name__ == "__main__":
     path = 'E:\\Users\A\Pictures\Pictures\lfw'
     work_dir = os.path.abspath(path)  # 图片放置目录
 
     sqliteName = os.path.join(work_dir, "photo.db")  # sqlite数据库文件
 
     # 初始化数据库表
     CreatePhotoDB(sqliteName)
 
     # 扫描目录,刷新数据库
 
     scanDirSearchFile(work_dir, sqliteName)

第二部：界面+索引比对：

 import sys
 import pandas as pd
 from PIL import Image
 import matplotlib.pyplot as plt
 import sqlite3
 from PyQt5 import QtWidgets, QtCore
 from PyQt5.QtGui import  *
 from PyQt5.QtWidgets import *
 from PyQt5.QtCore import*
 from PyQt5 import QtCore
 dbpath = 'E:\\Users\A\Pictures\Pictures\lfw\\photo.db'
 conn = sqlite3.connect(dbpath)
 class MainWindow(QWidget):
     def __init__(self,parent=None):
         super(MainWindow,self).__init__(parent)
         self.status = 0
         self.initUi(MainWindow)
         self.layout = QGridLayout(self)
         self.setLayout(self.layout)
 
     def initUi(self,MainWindow):
         self.createGridGroupBox()
         self.creatFormGroupBox()
 
         mainLayout = QVBoxLayout()
         hboxLayout = QHBoxLayout()
         # hboxLayout.addStretch()/拉升上面空间
         hboxLayout.addWidget(self.gridGroupBox)
         mainLayout.addLayout(hboxLayout)
         mainLayout.addWidget(self.formGroupBox)
         self.setLayout(mainLayout)
 
     @pyqtSlot(bool)
     def on_btn_open_clicked(self):
         self.status = 1
         self.filename = QFileDialog.getOpenFileName(self, "OpenFile", ".",
                                                     "Image Files(*.jpg *.jpeg *.png)")[0]
         # filename = self.filename
         print(self.filename)
         print(type(self.filename))
         print('status:',self.status)
         self.createGridGroupBox()
         self.picture_R(self.filename)
         # self.initUi()
         # print(self.createGridGroupBox(filename))
 
         return self.status,self.filename
 
     def createGridGroupBox(self):
         self.gridGroupBox = QGroupBox("Demo")
         layout = QGridLayout()
         nameLabel = QLabel("picture_name")
         self.nameLineEdit = QLineEdit("")
         layout.addWidget(nameLabel, 1, 0)
         layout.addWidget(self.nameLineEdit, 1, 1)
 
         imgeLabe1 = QLabel()
         imgeLabe2 = QLabel()
         # while(true):
         pixMap = None
 
         layout.setSpacing(10)
         layout.addWidget(imgeLabe1, 0, 2, 4, 1)
         layout.update()
         self.gridGroupBox.setLayout(layout)
         self.setWindowTitle('Search picture')
 
     def creatFormGroupBox(self):
         self.formGroupBox = QGroupBox("Form layout")
         layout = QFormLayout()
 
         serchBtn = QPushButton('查找')
 
         # manage_Btn.clicked.connect(manage_ui.manage)
         serchBtn.clicked.connect(self.name_link)
         # tphoBtn.clicked.connect(realtime_facenet_git.detectface)
 
         imgeLabe2 = QLabel()
         pixMap = QPixmap("fileName1")
         imgeLabe2.setPixmap(pixMap)
         layout.addWidget(imgeLabe2)
         hbox = QHBoxLayout()
         hbox.addWidget(serchBtn)
 
         vbox = QVBoxLayout()
         vbox.addLayout(hbox)
 
         self.formGroupBox.setLayout(vbox)
     def getName(self):
         name = self.nameLineEdit.text()  # 获取文本框内容
 
         df = ('文件名: %s \n ' % (name))
         print(df)
         return name
     def name_link(self):
 
         cursor = conn.execute("SELECT photoname,photoaddress from photoaddress")
         filename = self.getName()
         # filename = "202599.jpg"
         for row in cursor:
             # print(row[0])
             if row[0] == filename:
                 print(row[1])
 
                 img = Image.open(row[1])
                 plt.figure("Results")
                 plt.imshow(img)
                 plt.show()
 if __name__ == '__main__':
     app = QApplication(sys.argv)
     ex = MainWindow()
     ex.show()
     sys.exit(app.exec_())