compute_figure.py 文件源码

def plot_figure_factorization_b(images, structures, structure_optimized, rigidity_refined):
    # Figure 91
    PTH='./figure_factorization/'
    if not os.path.isdir(PTH):
        os.makedirs(PTH)

    io.imsave(PTH+'image_00.png',images[0])
    io.imsave(PTH+'image_01.png',images[1])
    io.imsave(PTH+'image_02.png',images[2])

    # Structure maps

    structure_min = np.percentile(structure_optimized[rigidity_refined==1].ravel(), 2)
    structure_max = np.percentile(structure_optimized[rigidity_refined==1].ravel(), 98)

    Is_bwd = structure2image(structures[0], rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)
    Is_fwd = structure2image(structures[1], rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)
    Is_comb = structure2image(structure_optimized, rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)

    io.imsave(PTH+'structure_bwd.png', Is_bwd)
    io.imsave(PTH+'structure_fwd.png', Is_fwd)
    io.imsave(PTH+'structure_comb.png', Is_comb)