OpenCV指纹识别实现代码实例

import os  
import cv2
def getNum(src, model):  
    # 读取两个指纹图像  
    img1 = cv2.imread(src)  
    img2 = cv2.imread(model)  
      
    # 创建 SIFT 特征检测器  
    sift = cv2.SIFT_create()  
      
    # 检测特征点和计算特征描述符  
    kp1, des1 = sift.detectAndCompute(img1, None)  
    kp2, des2 = sift.detectAndCompute(img2, None)  
      
    # 创建 FLANN 匹配器  
    flann = cv2.FlannBasedMatcher()  
      
    # 使用 KNN 算法找到最佳的两个匹配项  
    matches = flann.knnMatch(des1, des2, k=2)  
      
    # 存储好的匹配项  
    ok = []  
    for m, n in matches:  
        # 根据 Lowe's ratio test 过滤匹配项  
        if m.distance < 0.8 * n.distance:  
            ok.append(m)  
      
    # 返回好的匹配项的数量  
    num = len(ok)  
    return num

def getID(src, database):  
    max_num = 0  # 初始化最大匹配点数为0  
    for file in os.listdir(database):  # 遍历数据库中的文件  
        model = os.path.join(database, file)  # 构建模型文件的完整路径  
        num = getNum(src, model)  # 计算当前模型与源指纹的匹配点数  
        print("文件名：", file, "距离:", num)  # 打印文件名和匹配点数  
          
        # 如果当前匹配点数大于最大匹配点数，则更新最大匹配点数和对应的文件名  
        if num > max_num:  
            max_num = num  
            name = file  
      
    # 从文件名中提取ID（这里假设文件名的第一个字符是ID）  
    ID = name[0] if name else None  # 如果name为空，则ID为None（这里应该添加错误处理）  
      
    # 如果最大匹配点数小于100，则将ID设置为9999（这通常不是一个好的做法，因为它可能导致混淆）  
    if max_num < 100 and ID is not None:  # 添加ID非空的检查  
        ID = 9999  
      
    return ID

def getName(ID):  
    # 定义一个字典来映射ID到姓名  
    nameID = {0: 'a', 1: 'b', 2: 'c', 3: 'd', 4: 'e', 5: 'f',  
              6: 'g', 7: 'h', 8: 'i', 9: 'j', 9999: 'k'}  
      
    # 从字典中获取姓名（如果ID不在字典中，则返回None）  
    name = nameID.get(int(ID))  
    return name

if __name__ == "__main__":  
    src = 'src.bmp'  # 源指纹图像的路径  
    database = 'database'  # 指纹数据库目录的路径  
    ID = getID(src, database)  # 获取指纹ID  
    name = getName(ID)  # 根据ID获取姓名  
    print('识别结果：', name)  # 打印识别结果