蠡測 2022-07-12 23:05 采纳率: 57.1%
浏览 95
已结题

opencv以图识图

问题遇到的现象和发生背景

代码出自https://blog.csdn.net/cliukai/article/details/102467502
不知道哪些路径需要改动
求可以讲解下整个代码
csv是自己需要创建吗
cover的源图片是如何放入的

问题相关代码,请勿粘贴截图
import cv2
import numpy as np
import csv
import glob

class CoverDescriptor:
    def describe(self,image):
        #用BRISK方法对图片提取特征值
        descriptor = cv2.BRISK_create()
        #提取关键点和描述子,None对全图进行
        (kps,descs) = descriptor.detectAndCompute(image, None)
        kps = np.float32([kp.pt for kp in kps])
        return (kps,descs)

class CoverMatcher:
    def __init__(self,descriptor,coverPaths,ratio=0.7,minMatches=40,
                 useingHamming = True):
        self.descriptor = descriptor
        self.coverPaths = coverPaths
        self.ratio = ratio
        self.minMatcheers = minMatches
        if useingHamming:
            self.minMatcheers += "-Hamming"

    def search(self,queryKps,queryDescs):
        results = {}
        for coverPath in self.coverPaths:
            cover = cv2.imread(coverPath)
            gray = cv2.cvtColor(cover,cv2.COLOR_BGR2GRAY)
            (kps,descs) = self.descriptor.describle(gray)
            score = self.match(queryKps,queryDescs,kps,descs)
            results[coverPath] = score

        if len(results) > 0:
            results = sorted([(v,k) for (k,v) in results.items() if v > 0],
                             reverse = True)
            return results

    def match(self,kpsA,featuresA,kpsB,featuresB):
        matchers = cv2.DescriptorMatcher_create(self.distanceMethod)
        rawMatches = matcher.knnMatch(featuresB,featuresA,2)
        matches = []

        for m in rawMatches:
            if len(m) == 2 and m[0].distance < m[1].distance * self.ratio:
                matches.append((m[0].trainIdx,m[0].queryIdx))

        if len(marches) > self.minMatches:
            ptsA = np.float32([kpsA[i] for (i,_) in matches])
            ptsB = np.float32([kpsB[j] for (_,j) in matches])
            (_,status) = cv2.findHomography(ptsA,ptsB,cv2.RANSAC,4.0)
            return float(status.sum()) / status.size
        return -1.0


coverPath = r"C:\Users\蠡测\PycharmProjects\pythonProject1\test\source\*.jpg"
queryPath = r"C:\Users\蠡测\PycharmProjects\pythonProject1\test\search\search1.jpg"

bookDatabasePath = "books.csv"
bokkDateebase = {}
for book in csv.reader(open(bookDatabasePath)):
    bookDatabasePath[book[0]] = book[1:]

cd = CoverDescriptor()
cm = CoverDescriptor(cd,glob.glob(coverPath+"/*.jpg"),ratio=0.7,minMatches=40)
#加载查询图像
queryImage = cv2.imread(queryPath)
# 降噪,转换成灰度图
gray = cv2.cvtColor(queryImage, cv2.COLOR_BGR2GRAY)
# 提取关键点和描述子
(queryKps, queryDescs) = cd.describe(gray)
results = cm.search(queryKps,queryDescs)

#显示待查询的封面
cv2.imshow("query",queryImage)

#如果在数据库里没有找到书的封面
if len(results)==0:
    print("该书暂未收录")
    cv2.waitKey(0)
else:
    for i,(score,coverPath) in enumerate(results):
        (auther,title) = bookDatabase[coverPath[coverPath.rfind("\\")+1:]]
        print("loop{} precision is{:.2f}%: author is {} - title is {}".format(i+1,score*100,author,title))
        result = cv2.imread(coverPath)
        cv2.imshow("results",result)
        cv2.waitKey(0)

cv2.destroyAllWindows()

运行结果及报错内容

img

  • 写回答

3条回答 默认 最新

  • 爱晚乏客游 2022-07-13 09:13
    关注

    别的不说,先把你的中文路径去掉再说,opencv不支持中文路径的图片读写

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 7月21日
  • 已采纳回答 7月13日
  • 赞助了问题酬金10元 7月12日
  • 创建了问题 7月12日

悬赏问题

  • ¥20 如何解决访问网址被屏蔽的情况
  • ¥15 本地运行的vue2项目,使用内网穿透,前端界面一直转圈圈
  • ¥20 powerbuilder datawindow控件导出Excel数据,可不可以不自动覆盖原数据,而是在后面新插入入数据。
  • ¥100 无轴承永磁同步电机控制
  • ¥15 eps里添加本地倾斜模型
  • ¥15 telegram 问题
  • ¥15 nrf52810-c三个a 程序
  • ¥50 求一个半透明没有锯齿的圆角窗体的实现例子
  • ¥15 STM32cubeMX里的FreeRTOS无法释放内存
  • ¥15 CATIA有些零件打开直接单机确定终止