def extractSift(input_files): print "extracting Sift features" all_features_dict = {} for i, fname in enumerate(input_files): features_fname = fname + '.sift' if exists(features_fname) == False: print "calculating sift features for", fname sift.process_image(fname, features_fname) print "gathering sift features for", fname, locs, descriptors = sift.read_features_from_file(features_fname) os.unlink(features_fname) print descriptors.shape all_features_dict[fname] = descriptors return all_features_dict
def __extract_features_from_image_file(image_file, sift_tmp_file="tmp.sift"): sift.process_image(image_file, sift_tmp_file) _, descriptors = sift.read_features_from_file(sift_tmp_file) unlink(sift_tmp_file) return descriptors