# -*- coding: utf-8 -*- import pickle from PCV.localdescriptors import sift from PCV.imagesearch import imagesearch from PCV.geometry import homography from PCV.tools.imtools import get_imlist #载入图像列表 imlist = get_imlist('oxbuild/') nbr_images = len(imlist) #载入特征列表 featlist = [imlist[i][:-3]+'sift' for i in range(nbr_images)] #载入词汇 with open('oxbuild/vocabulary.pkl', 'rb') as f: voc = pickle.load(f) src = imagesearch.Searcher('testImaAdd.db',voc) #查询图像索引和查询返回的图像数 q_ind = 892 nbr_results = 20 # 常规查询(按欧式距离对结果排序) res_reg = [w[1] for w in src.query(imlist[q_ind])[:nbr_results]] print ('top matches (regular):', res_reg) #载入查询图像特征 q_locs,q_descr = sift.read_features_from_file(featlist[q_ind]) fp = homography.make_homog(q_locs[:,:2].T) #用单应性进行拟合建立RANSAC模型 model = homography.RansacModel() rank = {} #载入候选图像的特征 for ndx in res_reg[1:]: locs,descr = sift.read_features_from_file(featlist[ndx]) # get matches matches = sift.match(q_descr,descr) ind = matches.nonzero()[0] ind2 = matches[ind] tp = homography.make_homog(locs[:,:2].T) try: H,inliers = homography.H_from_ransac(fp[:,ind],tp[:,ind2],model,match_theshold=4) except: inliers = [] # store inlier count rank[ndx] = len(inliers) sorted_rank = sorted(rank.items(), key=lambda t: t[1], reverse=True) res_geom = [res_reg[0]]+[s[0] for s in sorted_rank] print ('top matches (homography):', res_geom) # 显示查询结果 imagesearch.plot_results(src,res_reg[:8]) #常规查询 imagesearch.plot_results(src,res_geom[:8]) #重排后的结果
时间: 2024-04-21 16:29:17 浏览: 146
zlib-1.2.3.7.tar.gz
这段代码是一个图像检索的例子,其中使用了SIFT算法提取图像的关键点和特征描述符,使用了单应性矩阵和RANSAC算法进行匹配和筛选,使用了词汇表和倒排索引技术来加速查询。具体来说,代码中首先载入了图像列表和特征列表,然后载入了之前训练好的词汇表,接着创建了一个图像搜索器Searcher。在进行查询时,首先使用常规查询(按欧式距离对结果排序)来获取最初的查询结果,然后使用单应性矩阵和RANSAC算法对查询图像和候选图像进行匹配和筛选,得到重排后的查询结果。最后使用plot_results函数来展示查询结果。
阅读全文