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'));
示例#2
0
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
示例#3
0
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'))