matcher

feature_matcher.py

+import argparse
+import numpy as np
+import cv2
+import h5py
+import os
+
+def save_h5(dict_to_save, filename):
+	with h5py.File(filename, 'w') as f:
+		for key in dict_to_save:
+			f.create_dataset(key, data=dict_to_save[key])
+			
+def load_h5(filename):
+	dict = {}
+	with h5py.File(filename, 'r') as f:
+		keys = [key for key in f.keys()]
+		for key in keys:
+			dict[key] = f[key][()]
+
+
+def compute_matches(desc1, desc2, kps1=None, kps2=None):
+    bf = cv2.BFMatcher(cv2.NORM_L2, crossCheck=True)
+    matches = bf.match(desc1, desc2)
+    return matches
+    
+if __name__ == '__main__'
+    parser = argparse.ArgumentParser(description='Feature extractor SIFT')
+    #parser.add_argument('--path', default='./dataset', type=str)
+    parser.add_argument('--desc', type=str)
+
+
+    arg = parser.parse_args()
+    print(arg) 
+    
+    descriptors = load_h5(desc)
+    
+    
+    for 
+    
+    
+