compute_figure.py 文件源码

def plot_figure_95(images, rigidity, structure, flow_init, flow):
    # Results figure for video.

    PTH='./figure_results/'
    if not os.path.isdir(PTH):
        os.makedirs(PTH)

    # Save frame triplet
    io.imsave(PTH+'image_0.png', images[0])
    io.imsave(PTH+'image_1.png', images[1])
    io.imsave(PTH+'image_2.png', images[2])

    I_rigidity = np.dstack((rigidity,rigidity,rigidity)).astype('float')
    I_rigidity[:,:,1] = 0
    I_rigidity[:,:,2] = 1-I_rigidity[:,:,2]
    io.imsave(PTH+'rigidity.png', I_rigidity)

    I_structure = structure2image(structure, rigidity)
    io.imsave(PTH+'structure.png', I_structure)

    I_mrflow = flow_viz.computeFlowImage(flow[0],flow[1])
    I_discreteflow = flow_viz.computeFlowImage(flow_init[0],flow_init[1])
    io.imsave(PTH+'mrflow.png', I_mrflow)
    io.imsave(PTH+'discreteflow.png', I_discreteflow)