def saveCorrespondingDirs(meta_dirs_image,meta_dirs_flo,sub_dirs_file,out_file_correspondences): image_dirs = getMovedDirPath(meta_dirs_image[0],meta_dirs_image[1],sub_dirs_file); flo_dirs = getMovedDirPath(meta_dirs_flo[0],meta_dirs_flo[1],sub_dirs_file); flo_idx=[]; image_dirs_stripped=[dir_curr.rsplit('/',2)[1] for dir_curr in image_dirs]; flo_dirs_stripped=[dir_curr.rsplit('/',2)[1] for dir_curr in flo_dirs]; index_array= util.getIndexingArray(flo_dirs_stripped,image_dirs_stripped); flo_dirs=np.array(flo_dirs); flo_dirs=flo_dirs[index_array]; flo_dirs=list(flo_dirs); pickle.dump(zip(image_dirs,flo_dirs),open(out_file_correspondences,'wb'));
def sortBySmallestDistanceRank(indices_interest,neighbor_index): indices_interest_index=util.getIndexingArray(neighbor_index,indices_interest); indices_interest_index=np.array(indices_interest_index); idx_to_order=np.argsort(indices_interest_index); indices_interest_index=indices_interest_index[idx_to_order]; indices_ordered=indices_interest[idx_to_order]; nn_rank=neighbor_index[indices_interest_index] return indices_interest_index,idx_to_order
def saveCorrespondingDirs(meta_dirs_image, meta_dirs_flo, sub_dirs_file, out_file_correspondences): image_dirs = getMovedDirPath(meta_dirs_image[0], meta_dirs_image[1], sub_dirs_file) flo_dirs = getMovedDirPath(meta_dirs_flo[0], meta_dirs_flo[1], sub_dirs_file) flo_idx = [] image_dirs_stripped = [ dir_curr.rsplit('/', 2)[1] for dir_curr in image_dirs ] flo_dirs_stripped = [dir_curr.rsplit('/', 2)[1] for dir_curr in flo_dirs] index_array = util.getIndexingArray(flo_dirs_stripped, image_dirs_stripped) flo_dirs = np.array(flo_dirs) flo_dirs = flo_dirs[index_array] flo_dirs = list(flo_dirs) pickle.dump(zip(image_dirs, flo_dirs), open(out_file_correspondences, 'wb'))