extract orb

extract_orb.py

+import argparse
+import numpy as np
+import cv2
+import h5py
+import os
+from utilities import *
+
+parser = argparse.ArgumentParser(description='Feature extractor SIFT')
+parser.add_argument('--path', default='./dataset', type=str)
+
+arg = parser.parse_args()
+print(arg)
+
+
+orb = cv2.ORB_create()
+
+kp_dict = {}
+desc_dict = {}
+
+for image_name in os.listdir(arg.path):
+	im = cv2.imread(arg.path + image_name)
+	gray= cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
+	kp, descriptors = orb.detectAndCompute(gray,None)
+	#print(kp)
+	#print(descriptors)
+	keypoints = np.array([k.pt for k in kp])
+	kp_dict[image_name] = keypoints
+	desc_dict[image_name] = descriptors
+	
+
+save_h5(kp_dict, 'keypoints_orb.h5')
+save_h5(desc_dict, 'descriptors_orb.h5')
+
+k2 = load_h5('keypoints_orb.h5')
+d2 = load_h5("descriptors_orb.h5")
+
+keypoints = k2
+descriptors = d2
+