compute_figure.py 文件源码

def plot_figure_video_pasted_example(rigidity, flow_discrete, flow_ours):
    # Figure 94
    PTH='./figure_pasted/'
    if not os.path.isdir(PTH):
        os.makedirs(PTH)

    I_rigidity = np.dstack((rigidity,rigidity,rigidity)).astype('float')
    I_df = flow_viz.computeFlowImage(flow_discrete[0],flow_discrete[1])
    I_struc = flow_viz.computeFlowImage(flow_ours[0],flow_ours[1])

    I_struc_filtered = I_rigidity*I_struc

    I_final = I_struc_filtered + (1-I_rigidity)*I_df

    I_rigidity_ = I_rigidity.copy()
    I_rigidity_[:,:,1] = 0
    I_rigidity_[:,:,2] = 1-I_rigidity_[:,:,2]

    io.imsave(PTH+'rigidiyt.png', I_rigidity_)
    io.imsave(PTH+'discreteflow.png', I_df)
    io.imsave(PTH+'structureflow.png', I_struc)
    io.imsave(PTH+'structureflow_filtered.png', I_struc_filtered.astype('uint8'))
    io.imsave(PTH+'mrflow.png', I_final.astype('uint8'))