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