process_cities.py 文件源码

def vis_pairs(prms, isSave=False, svIdx=None, svPath=None):
    imName1, imName2, euls, trans = read_pairs(prms)
    N = len(imName1)
    seed      = 3
    randState = np.random.RandomState(seed)
    perm = randState.permutation(N)
    fig = plt.figure()
    plt.ion()
    imName1 = [imName1[i] for i in perm]
    imName2 = [imName2[i] for i in perm]
    euls    = [euls[i] for i in perm]
    trans   = [trans[i] for i in perm]
    titleStr = 'Trans: ' + '%.3f ' * 3 + 'Rot: ' + '%.3f ' * 3
    count   = 0
    numSave = 0
    for (im1,im2,eu,tr) in zip(imName1, imName2, euls, trans):
        titleName = titleStr % (tuple(tr) + eu)
        im1 = scm.imread(im1)
        im2 = scm.imread(im2)
        print count
        if isSave:
            if count in svIdx:
                imN1 = svPath % (count,1)
                imN2 = svPath % (count,2)
                scm.imsave(imN1,im1)
                scm.imsave(imN2,im2)
                numSave += 1
                if numSave==len(svIdx):
                    return 
        else:
            vu.plot_pairs(im1, im2, fig, titleStr=titleName)
            cmd = raw_input()   
            if cmd == 'exit':
                return
        count += 1
##