compute_figure.py 文件源码

def plot_figure_3(image, rigidity_cnn, rigidity_motion, rigidity_structure, rigidity_refined):
    if not os.path.isdir('./rigidityestimation'):
        os.makedirs('./rigidityestimation')

    cm_bwr = plt.get_cmap('bwr')
    Irigidity_cnn = cm_bwr(rigidity_cnn.astype('float32'))
    Irigidity_motion = cm_bwr(rigidity_motion.astype('float32'))
    Irigidity_structure = cm_bwr(rigidity_structure.astype('float32'))
    Irigidity_refined = cm_bwr(rigidity_refined.astype('float32'))

    io.imsave('./rigidityestimation/01_image.png', img_as_ubyte(image))
    io.imsave('./rigidityestimation/02_rigidity_cnn.png', Irigidity_cnn)
    io.imsave('./rigidityestimation/03_rigidity_motion.png', Irigidity_motion)
    io.imsave('./rigidityestimation/04_rigidity_structure.png', Irigidity_structure)
    io.imsave('./rigidityestimation/05_rigidity_refined.png', Irigidity_refined)