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
