-
Notifications
You must be signed in to change notification settings - Fork 2
/
detector_vo.py
35 lines (28 loc) · 812 Bytes
/
detector_vo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import numpy as np
import cv2
def detector(a):
if (a == 1):
filename = "orb.csv"
detector = cv2.ORB_create(nfeatures=1500)
pose = "orb.txt"
if (a == 2):
filename = "surf.csv"
detector = cv2.xfeatures2d.SURF_create()
pose = "surf.txt"
if (a == 3):
filename = "sift.csv"
detector = cv2.xfeatures2d.SIFT_create()
pose = "sift.txt"
if (a == 4):
filename = "kaze.csv"
detector = cv2.KAZE_create()
pose = "kaze.txt"
if (a == 5):
filename = "akaze.csv"
detector = cv2.AKAZE_create()
pose = "akaze.txt"
if (a == 6):
filename = "brisk.csv"
detector = cv2.BRISK_create()
pose = "brisk.txt"
return (detector, filename, pose)