对图片进行索引，存入数据库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_())

对图片进行索引，存入数据库sqlite3中，实现快速搜索打开的相关教程结束。