generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

correct unknown feature report in find_obj.py

Browse Source
This commit is contained in:
Alexander Mordvintsev 2012-07-16 12:31:56 +00:00
parent 3ce5b01543
commit ad7a6ec41f
1 changed files with 2 additions and 1 deletions

3
samples/python2/find_obj.py

@ -19,7 +19,6 @@ FLANN_INDEX_LSH = 6
def init_feature(name): def init_feature(name):
detector, matcher = None, None
chunks = name.split('-') chunks = name.split('-')
if chunks[0] == 'sift': if chunks[0] == 'sift':
detector = cv2.SIFT() detector = cv2.SIFT()
@ -30,6 +29,8 @@ def init_feature(name):
elif chunks[0] == 'orb': elif chunks[0] == 'orb':
detector = cv2.ORB(400) detector = cv2.ORB(400)
norm = cv2.NORM_HAMMING norm = cv2.NORM_HAMMING
else:
return None, None
if 'flann' in chunks: if 'flann' in chunks:
if norm == cv2.NORM_L2: if norm == cv2.NORM_L2:
flann_params = dict(algorithm = FLANN_INDEX_KDTREE, trees = 5) flann_params = dict(algorithm = FLANN_INDEX_KDTREE, trees = 5)