1. 提取特征保存到本地,以减少重复计算
import numpy as np
import cv2
import time
import os
if __name__ == '__main__':
db_path = './db/'
save_path='./db_des/'
r=128 #resize 后的大小
ratio=0.76
sift = cv2.xfeatures2d.SURF_create()
# 从搜索库中进行计算
sort_list = []
for imgname in os.listdir(db_path):
#图像2
img = cv2.imread(db_path+imgname)
img = cv2.resize(img, (r, r), interpolation=cv2.INTER_CUBIC)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # 灰度处理图像
kp, des = sift.detectAndCompute(img, None) # des是描述子
np.save(save_path+imgname+'.npy',des)
2. 待检索图片检索分类
'''
商品ID:商品名称
dict_label={'01': '削皮刀',
'02': '婴儿湿巾',
&