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)
评论列表
文章目录