#?-*-?coding:utf-8?-*-
import?cv2
import?numpy?as?np
from?os.path?import?join

trainpath?=?‘D:\PyCharm?2018.2.2\program\image\CarData\TrainImages‘

#定義函數(shù)返回訓(xùn)練集圖像的路徑
def?path（clsi）:
????return?‘%s\%s%d.pgm‘%（trainpathclsi+1）

posneg?=?‘pos-‘‘neg-‘

#創(chuàng)建兩個(gè)SIFT對(duì)象，一個(gè)用來(lái)提取關(guān)鍵點(diǎn)，一個(gè)用來(lái)提取特征描述符
sift_kp?=?cv2.xfeatures2d.SIFT_create（）
sift_des?=?cv2.xfeatures2d.SIFT_create（）

#創(chuàng)建FLANN匹配器
indexParams?=?dict（algorithm=1?trees=5）
searchParams?=?{}
flann?=?cv2.FlannbasedMatcher（indexParams?searchParams）

#創(chuàng)建bow訓(xùn)練器，指定簇?cái)?shù)為40
bow_kmeans_trainer?=?cv2.BOWKMeansTrainer（40）
#創(chuàng)建bow特征提取器，視覺(jué)詞匯將作為bow提取器的輸入，bow提取器用來(lái)提取圖像的特征，與視覺(jué)單詞進(jìn)行特征匹配，返回以視覺(jué)單詞為詞基的向量特征
bow_extractor?=?cv2.BOWImgDescriptorExtractor（sift_desflann）

#函數(shù)返回圖像的特征描述符，SIFT.compute（圖像，關(guān)鍵點(diǎn)）
def?des_extract（imgpath）:
????img?=?cv2.imread（imgpath0）
????return?sift_des.compute（imgsift_kp.detect（img））[1]

#將正負(fù)樣本的描述符添加至bow訓(xùn)練器中
for?i?in?range（50）:
????bow_kmeans_trainer.add（des_extract（path（posi）））
????bow_kmeans_trainer.add（des_extract（path（negi）））

#cluster（）函數(shù)執(zhí)行k-means分類(lèi)，并返回視覺(jué)詞匯，將視覺(jué)詞匯輸入到bow提取器中
vocabulary?=?bow_kmeans_trainer.cluster（）
bow_extractor.setVocabulary（vocabulary）

#定義函數(shù)，返回bow提取器提取的特征描述符
def?bow_features（imgpath）:
????img?=?cv2.imread（imgpath）
????return?bow_extractor.compute（img