蠡測 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 完全没有学习过GAN,看了CSDN的一篇文章,里面有代码但是完全不知道如何操作
  • ¥15 使用ue5插件narrative时如何切换关卡也保存叙事任务记录
  • ¥20 软件测试决策法疑问求解答
  • ¥15 win11 23H2删除推荐的项目,支持注册表等
  • ¥15 matlab 用yalmip搭建模型,cplex求解,线性化处理的方法
  • ¥15 qt6.6.3 基于百度云的语音识别 不会改
  • ¥15 关于#目标检测#的问题:大概就是类似后台自动检测某下架商品的库存,在他监测到该商品上架并且可以购买的瞬间点击立即购买下单
  • ¥15 神经网络怎么把隐含层变量融合到损失函数中?
  • ¥15 lingo18勾选global solver求解使用的算法
  • ¥15 全部备份安卓app数据包括密码,可以复制到另一手机上运行