compute_figure.py 文件源码

def plot_figure_2(images,
        flow_init,
        rigidity_init,
        structure_init,
        occlusions,
        rigidity_refined,
        structure_refined,
        flow_estimated):
    if not os.path.isdir('./diagram'):
        os.makedirs('diagram')

    io.imsave('./diagram/inputframe0.png', img_as_ubyte(images[0]))
    io.imsave('./diagram/inputframe1.png', img_as_ubyte(images[1]))
    io.imsave('./diagram/inputframe2.png', img_as_ubyte(images[2]))

    Iuvinit_bwd = flow_viz.computeFlowImage(flow_init[0][0],flow_init[0][1])
    Iuvinit_fwd = flow_viz.computeFlowImage(flow_init[1][0],flow_init[1][1])
    io.imsave('./diagram/inputflow0.png', Iuvinit_bwd)
    io.imsave('./diagram/inputflow1.png', Iuvinit_fwd)

    cm_bwr = plt.get_cmap('bwr')
    Irigidity_init = cm_bwr(rigidity_init.astype('float32'))
    Irigidity_refined = cm_bwr(rigidity_refined.astype('float32'))
    io.imsave('./diagram/rigidity_init.png', Irigidity_init)
    io.imsave('./diagram/rigidity_refined.png', Irigidity_refined)

    Istructure_init = structure2image(structure_init, rigidity_refined)
    Istructure_refined = structure2image(structure_refined, rigidity_refined)
    io.imsave('./diagram/structure_init.png', Istructure_init)
    io.imsave('./diagram/structure_refined.png', Istructure_refined)

    occ_bwd, occ_fwd = occlusions
    Iocclusions = np.ones_like(Istructure_init) * 255
    Iocclusions[:,:,0][occ_bwd>0] = 255
    Iocclusions[:,:,1][occ_bwd>0] = 0
    Iocclusions[:,:,2][occ_bwd>0] = 0
    Iocclusions[:,:,0][occ_fwd>0] = 0
    Iocclusions[:,:,1][occ_fwd>0] = 0
    Iocclusions[:,:,2][occ_fwd>0] = 255
    io.imsave('./diagram/occlusions.png', Iocclusions)

    Iuvest = flow_viz.computeFlowImage(flow_estimated[0],flow_estimated[1])
    io.imsave('./diagram/outputflow.png', Iuvest)