def visualize_jacobi(phi, spacing, img=None, file_path=None, visual=True):
"""
:param phi: Bxdimx X xYxZ
:param spacing: [sx,sy,sz]
:param img: Bx1xXxYxZ
:param file_path: saving path
:return:
"""
phi_sz = phi.shape
n_batch = phi_sz[0]
dim = phi_sz[1]
phi_np = utils.t2np(phi)
if img is not None:
assert phi.shape[0] == img.shape[0]
img_np = utils.t2np(img)
fd = fdt.FD_np(spacing)
dfx = fd.dXc(phi_np[:, 0, ...])
dfy = fd.dYc(phi_np[:, 1, ...])
dfz = 1.
if dim == 3:
dfz = fd.dZc(phi_np[:, 2, ...])
jacobi_det = dfx * dfy * dfz
jacobi_neg = np.ma.masked_where(jacobi_det >= 0, jacobi_det)
#jacobi_neg = (jacobi_det<0).astype(np.float32)
jacobi_abs = -np.sum(jacobi_det[jacobi_det < 0.]) #
jacobi_num = np.sum(jacobi_det < 0.)
if dim == 3:
print("print folds for each channel {},{},{}".format(
np.sum(dfx < 0.), np.sum(dfy < 0.), np.sum(dfz < 0.)))
print("the jacobi_value of fold points for current map is {}".format(
jacobi_abs))
print("the number of fold points for current map is {}".format(jacobi_num))
if visual:
for i in range(n_batch):
if dim == 2:
sp = 111
font = {'size': 10}
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
plt.subplot(sp).set_axis_off()
plt.imshow(utils.t2np(img_np[i, 0]))
plt.imshow(jacobi_neg[i], cmap='gray', alpha=1.)
plt.colorbar().ax.tick_params(labelsize=10)
plt.title('img_jaocbi', font)
if not file_path:
plt.show()
else:
plt.savefig(file_path, dpi=300)
plt.clf()
if dim == 3:
if file_path:
jacobi_abs_map = np.abs(jacobi_det)
jacobi_img = sitk.GetImageFromArray(jacobi_abs_map[i])
pth = os.path.join(file_path)
sitk.WriteImage(jacobi_img, pth)
def plot_2d_img(img, name, path=None):
"""
:param img: X x Y x Z
:param name: title
:param path: saving path
:param show:
:return:
"""
sp = 111
img = torch.squeeze(img)
font = {'size': 10}
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
plt.subplot(sp).set_axis_off()
plt.imshow(
utils.t2np(img)) #,vmin=0.0590, vmax=0.0604) #vmin=0.0590, vmax=0.0604
plt.colorbar().ax.tick_params(labelsize=10)
plt.title(name, font)
if not path:
plt.show()
else:
plt.savefig(path, dpi=300)
plt.clf()
def _show_current_images_3d(iS,
iT,
iW,
iSL,
iTL,
iWL,
iter,
vizImage,
vizName,
phiWarped,
visual_param=None,
i=0,
extraImage=None,
extraName=None):
if iSL is not None and iTL is not None:
phiw_a = 3 if phiWarped is not None else 2
if vizImage is None and extraImage is None:
fig, ax = plt.subplots(phiw_a + 4, 3)
iSL_a = phiw_a + 1
iTL_a = phiw_a + 2
iWL_a = phiw_a + 3
elif extraImage is None:
fig, ax = plt.subplots(phiw_a + 5, 3)
vizi_a = phiw_a + 1
iSL_a = phiw_a + 2
iTL_a = phiw_a + 3
iWL_a = phiw_a + 4
else:
# if is_lung:
# fig, ax = plt.subplots(9, 3,figsize=(6,15))
# else:
# fig, ax = plt.subplots(9, 3)
fig, ax = plt.subplots(phiw_a + 6, 3)
vizi_a = phiw_a + 1
ext_a = phiw_a + 2
iSL_a = phiw_a + 3
iTL_a = phiw_a + 4
iWL_a = phiw_a + 5
else:
if (phiWarped is not None) and (vizImage is not None):
ext_a = 5 if extraImage is not None else 4
fig, ax = plt.subplots(ext_a + 1, 3)
phiw_a = 3
vizi_a = 4
elif (phiWarped is not None):
ext_a = 4 if extraImage is not None else 3
fig, ax = plt.subplots(ext_a + 1, 3)
phiw_a = 3
elif (vizImage is not None):
ext_a = 4 if extraImage is not None else 3
fig, ax = plt.subplots(ext_a + 1, 3)
vizi_a = 3
else:
fig, ax = plt.subplots(3, 3)
plt.suptitle('Iteration = ' + str(iter))
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
#plt.subplots_adjust(top=0.99, bottom=0.01, hspace=0.15, wspace=0.04)
ivsx = viewers.ImageViewer3D_Sliced(ax[0][0], utils.t2np(iS), 0,
'source X', True)
ivsy = viewers.ImageViewer3D_Sliced(ax[0][1], utils.t2np(iS), 1,
'source Y', True)
ivsz = viewers.ImageViewer3D_Sliced(ax[0][2], utils.t2np(iS), 2,
'source Z', True)
ivtx = viewers.ImageViewer3D_Sliced(ax[1][0], utils.t2np(iT), 0,
'target X', True)
ivty = viewers.ImageViewer3D_Sliced(ax[1][1], utils.t2np(iT), 1,
'target Y', True)
ivtz = viewers.ImageViewer3D_Sliced(ax[1][2], utils.t2np(iT), 2,
'target Z', True)
ivwx = viewers.ImageViewer3D_Sliced(ax[2][0], utils.t2np(iW), 0,
'warped X', True)
ivwy = viewers.ImageViewer3D_Sliced(ax[2][1], utils.t2np(iW), 1,
'warped Y', True)
ivwz = viewers.ImageViewer3D_Sliced(ax[2][2], utils.t2np(iW), 2,
'warped Z', True)
if phiWarped is not None:
ivwxc = viewers.ImageViewer3D_Sliced_Contour(ax[phiw_a][0],
utils.t2np(iW),
utils.t2np(phiWarped), 0,
'warped X', True)
ivwyc = viewers.ImageViewer3D_Sliced_Contour(ax[phiw_a][1],
utils.t2np(iW),
utils.t2np(phiWarped), 1,
'warped Y', True)
ivwzc = viewers.ImageViewer3D_Sliced_Contour(ax[phiw_a][2],
utils.t2np(iW),
utils.t2np(phiWarped), 2,
'warped Z', True)
if vizImage is not None:
ivvxc = viewers.ImageViewer3D_Sliced(
ax[vizi_a][0], utils.lift_to_dimension(utils.t2np(vizImage), 3), 0,
vizName + ' X', True)
ivvyc = viewers.ImageViewer3D_Sliced(
ax[vizi_a][1], utils.lift_to_dimension(utils.t2np(vizImage), 3), 1,
vizName + ' Y', True)
ivvzc = viewers.ImageViewer3D_Sliced(
ax[vizi_a][2], utils.lift_to_dimension(utils.t2np(vizImage), 3), 2,
vizName + ' Z', True)
if extraImage is not None:
ivexc = viewers.ImageViewer3D_Sliced(
ax[ext_a][0], utils.lift_to_dimension(utils.t2np(extraImage), 3),
0, extraName + ' X', True)
iveyc = viewers.ImageViewer3D_Sliced(
ax[ext_a][1], utils.lift_to_dimension(utils.t2np(extraImage), 3),
1, extraName + ' Y', True)
ivezc = viewers.ImageViewer3D_Sliced(
ax[ext_a][2], utils.lift_to_dimension(utils.t2np(extraImage), 3),
2, extraName + ' Z', True)
if iSL is not None and iTL is not None:
ivslxc = viewers.ImageViewer3D_Sliced(
ax[iSL_a][0], utils.lift_to_dimension(utils.t2np(iSL), 3), 0,
'Lsource X', True, True)
ivslyc = viewers.ImageViewer3D_Sliced(
ax[iSL_a][1], utils.lift_to_dimension(utils.t2np(iSL), 3), 1,
'Lsource Y', True, True)
ivslzc = viewers.ImageViewer3D_Sliced(
ax[iSL_a][2], utils.lift_to_dimension(utils.t2np(iSL), 3), 2,
'Lsource Z', True, True)
ivtlxc = viewers.ImageViewer3D_Sliced(
ax[iTL_a][0], utils.lift_to_dimension(utils.t2np(iTL), 3), 0,
'LTarget X', True, True)
ivtlyc = viewers.ImageViewer3D_Sliced(
ax[iTL_a][1], utils.lift_to_dimension(utils.t2np(iTL), 3), 1,
'LTarget Y', True, True)
ivtlzc = viewers.ImageViewer3D_Sliced(
ax[iTL_a][2], utils.lift_to_dimension(utils.t2np(iTL), 3), 2,
'LTarget Z', True, True)
ivwlxc = viewers.ImageViewer3D_Sliced(
ax[iWL_a][0], utils.lift_to_dimension(utils.t2np(iWL), 3), 0,
'LWarped X', True, True)
ivwlyc = viewers.ImageViewer3D_Sliced(
ax[iWL_a][1], utils.lift_to_dimension(utils.t2np(iWL), 3), 1,
'LWarped Y', True, True)
ivwlzc = viewers.ImageViewer3D_Sliced(
ax[iWL_a][2], utils.lift_to_dimension(utils.t2np(iWL), 3), 2,
'LWarped Z', True, True)
feh = viewers.FigureEventHandler(fig)
feh.add_axes_event('button_press_event', ax[0][0], ivsx.on_mouse_press,
ivsx.get_synchronize, ivsx.set_synchronize)
feh.add_axes_event('button_press_event', ax[0][1], ivsy.on_mouse_press,
ivsy.get_synchronize, ivsy.set_synchronize)
feh.add_axes_event('button_press_event', ax[0][2], ivsz.on_mouse_press,
ivsz.get_synchronize, ivsz.set_synchronize)
feh.add_axes_event('button_press_event', ax[1][0], ivtx.on_mouse_press,
ivtx.get_synchronize, ivtx.set_synchronize)
feh.add_axes_event('button_press_event', ax[1][1], ivty.on_mouse_press,
ivty.get_synchronize, ivty.set_synchronize)
feh.add_axes_event('button_press_event', ax[1][2], ivtz.on_mouse_press,
ivtz.get_synchronize, ivtz.set_synchronize)
feh.add_axes_event('button_press_event', ax[2][0], ivwx.on_mouse_press,
ivwx.get_synchronize, ivwx.set_synchronize)
feh.add_axes_event('button_press_event', ax[2][1], ivwy.on_mouse_press,
ivwy.get_synchronize, ivwy.set_synchronize)
feh.add_axes_event('button_press_event', ax[2][2], ivwz.on_mouse_press,
ivwz.get_synchronize, ivwz.set_synchronize)
if phiWarped is not None:
feh.add_axes_event('button_press_event', ax[phiw_a][0],
ivwxc.on_mouse_press, ivwxc.get_synchronize,
ivwxc.set_synchronize)
feh.add_axes_event('button_press_event', ax[phiw_a][1],
ivwyc.on_mouse_press, ivwyc.get_synchronize,
ivwyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[phiw_a][2],
ivwzc.on_mouse_press, ivwzc.get_synchronize,
ivwzc.set_synchronize)
if vizImage is not None:
feh.add_axes_event('button_press_event', ax[vizi_a][0],
ivvxc.on_mouse_press, ivvxc.get_synchronize,
ivvxc.set_synchronize)
feh.add_axes_event('button_press_event', ax[vizi_a][1],
ivvyc.on_mouse_press, ivvyc.get_synchronize,
ivvyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[vizi_a][2],
ivvzc.on_mouse_press, ivvzc.get_synchronize,
ivvzc.set_synchronize)
if extraImage is not None:
feh.add_axes_event('button_press_event', ax[ext_a][0],
ivexc.on_mouse_press, ivexc.get_synchronize,
ivexc.set_synchronize)
feh.add_axes_event('button_press_event', ax[ext_a][1],
iveyc.on_mouse_press, iveyc.get_synchronize,
iveyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[ext_a][2],
ivezc.on_mouse_press, ivezc.get_synchronize,
ivezc.set_synchronize)
if iSL is not None and iTL is not None:
feh.add_axes_event('button_press_event', ax[iSL_a][0],
ivslxc.on_mouse_press, ivslxc.get_synchronize,
ivslxc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iSL_a][1],
ivslyc.on_mouse_press, ivslyc.get_synchronize,
ivslyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iSL_a][2],
ivslzc.on_mouse_press, ivslzc.get_synchronize,
ivslzc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iTL_a][0],
ivtlxc.on_mouse_press, ivtlxc.get_synchronize,
ivtlxc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iTL_a][1],
ivtlyc.on_mouse_press, ivtlyc.get_synchronize,
ivtlyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iTL_a][2],
ivtlzc.on_mouse_press, ivtlzc.get_synchronize,
ivtlzc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iWL_a][0],
ivwlxc.on_mouse_press, ivwlxc.get_synchronize,
ivwlxc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iWL_a][1],
ivwlyc.on_mouse_press, ivwlyc.get_synchronize,
ivwlyc.set_synchronize)
feh.add_axes_event('button_press_event', ax[iWL_a][2],
ivwlzc.on_mouse_press, ivwlzc.get_synchronize,
ivwlzc.set_synchronize)
if iSL is not None and iTL is not None:
if vizImage is not None and extraImage is not None:
feh.synchronize([
ax[0][0], ax[1][0], ax[2][0], ax[phiw_a][0], ax[vizi_a][0],
ax[ext_a][0], ax[iSL_a][0], ax[iTL_a][0], ax[iWL_a][0]
])
feh.synchronize([
ax[0][1], ax[1][1], ax[2][1], ax[phiw_a][1], ax[vizi_a][1],
ax[ext_a][1], ax[iSL_a][1], ax[iTL_a][1], ax[iWL_a][1]
])
feh.synchronize([
ax[0][2], ax[1][2], ax[2][2], ax[phiw_a][2], ax[vizi_a][2],
ax[ext_a][2], ax[iSL_a][2], ax[iTL_a][2], ax[iWL_a][2]
])
elif vizImage is not None:
feh.synchronize([
ax[0][0], ax[1][0], ax[2][0], ax[phiw_a][0], ax[vizi_a][0],
ax[iSL_a][0], ax[iTL_a][0], ax[iWL_a][0]
])
feh.synchronize([
ax[0][1], ax[1][1], ax[2][1], ax[phiw_a][1], ax[vizi_a][1],
ax[iSL_a][1], ax[iTL_a][1], ax[iWL_a][1]
])
feh.synchronize([
ax[0][2], ax[1][2], ax[2][2], ax[phiw_a][2], ax[vizi_a][2],
ax[iSL_a][2], ax[iTL_a][2], ax[iWL_a][2]
])
else:
feh.synchronize([
ax[0][0], ax[1][0], ax[2][0], ax[phiw_a][0], ax[iSL_a][0],
ax[iTL_a][0], ax[iWL_a][0]
])
feh.synchronize([
ax[0][1], ax[1][1], ax[2][1], ax[phiw_a][1], ax[iSL_a][1],
ax[iTL_a][1], ax[iWL_a][1]
])
feh.synchronize([
ax[0][2], ax[1][2], ax[2][2], ax[phiw_a][2], ax[iSL_a][2],
ax[iTL_a][2], ax[iWL_a][2]
])
if (phiWarped is not None) and (vizImage is not None):
feh.synchronize(
[ax[0][0], ax[1][0], ax[2][0], ax[phiw_a][0], ax[vizi_a][0]])
feh.synchronize(
[ax[0][1], ax[1][1], ax[2][1], ax[phiw_a][1], ax[vizi_a][1]])
feh.synchronize(
[ax[0][2], ax[1][2], ax[2][2], ax[phiw_a][2], ax[vizi_a][2]])
elif (phiWarped is not None):
feh.synchronize([ax[0][0], ax[1][0], ax[2][0], ax[phiw_a][0]])
feh.synchronize([ax[0][1], ax[1][1], ax[2][1], ax[phiw_a][1]])
feh.synchronize([ax[0][2], ax[1][2], ax[2][2], ax[phiw_a][2]])
elif (vizImage is not None):
feh.synchronize([ax[0][0], ax[1][0], ax[2][0], ax[vizi_a][0]])
feh.synchronize([ax[0][1], ax[1][1], ax[2][1], ax[vizi_a][1]])
feh.synchronize([ax[0][2], ax[1][2], ax[2][2], ax[vizi_a][2]])
else:
feh.synchronize([ax[0][0], ax[1][0], ax[2][0]])
feh.synchronize([ax[0][1], ax[1][1], ax[2][1]])
feh.synchronize([ax[0][2], ax[1][2], ax[2][2]])
if visual_param is not None:
if i == 0 and visual_param['visualize']:
plt.show()
if visual_param['save_fig']:
file_name = visual_param['pair_name'][i]
join_p = lambda pth1, pth2: os.path.join(pth1, pth2)
make_dir(join_p(visual_param['save_fig_path_byname'], file_name))
make_dir(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']))
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byname'], file_name),
visual_param['iter'] + extension),
dpi=dpi)
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']), file_name + extension),
dpi=dpi)
plt.close('all')
else:
plt.show()
plt.clf()
def _show_current_images_2d_map(iS,
iT,
iW,
iSL,
iTL,
iWL,
iter,
vizImage,
vizName,
phiWarped,
visual_param=None,
i=0):
if iSL is not None and iTL is not None:
sp_s = 331
sp_t = 332
sp_w = 333
sp_c = 334
sp_p = 335
sp_v = 336
sp_ls = 337
sp_lt = 338
sp_lw = 339
elif (vizImage is not None) and (phiWarped is not None):
sp_s = 231
sp_t = 232
sp_w = 233
sp_c = 234
sp_p = 235
sp_v = 236
elif (vizImage is not None):
sp_s = 231
sp_t = 232
sp_w = 233
sp_c = 234
sp_v = 235
elif (phiWarped is not None):
sp_s = 231
sp_t = 232
sp_w = 233
sp_c = 234
sp_p = 235
else:
sp_s = 221
sp_t = 222
sp_w = 223
sp_c = 224
font = {'size': 10}
plt.suptitle('Iteration = ' + str(iter))
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
plt.subplot(sp_s).set_axis_off()
plt.imshow(utils.t2np(iS), cmap='gray')
plt.colorbar().ax.tick_params(labelsize=3)
plt.title('source image', font)
plt.subplot(sp_t).set_axis_off()
plt.imshow(utils.t2np(iT), cmap='gray')
plt.colorbar().ax.tick_params(labelsize=3)
plt.title('target image', font)
plt.subplot(sp_w).set_axis_off()
plt.imshow(utils.t2np(iW), cmap='gray')
plt.colorbar().ax.tick_params(labelsize=3)
plt.title('warped image', font)
plt.subplot(sp_c).set_axis_off()
plt.imshow(checkerboard_2d(utils.t2np(iW), utils.t2np(iT)), cmap='gray')
plt.colorbar().ax.tick_params(labelsize=3)
plt.title('checkerboard', font)
if phiWarped is not None:
plt.subplot(sp_p).set_axis_off()
plt.imshow(utils.t2np(iW), cmap='gray')
plt.contour(utils.t2np(phiWarped[0, :, :]),
np.linspace(-1, 1, 20),
colors='r',
linestyles='solid',
linewidths=0.5)
plt.contour(utils.t2np(phiWarped[1, :, :]),
np.linspace(-1, 1, 20),
colors='r',
linestyles='solid',
linewidths=0.5)
plt.colorbar().ax.tick_params(labelsize=3)
plt.title('warped image + grid', font)
if vizImage is not None:
plt.subplot(sp_v).set_axis_off()
plt.imshow(utils.lift_to_dimension(utils.t2np(vizImage), 2),
cmap='gray')
plt.colorbar().ax.tick_params(labelsize=3)
plt.title(vizName, font)
if iSL is not None and iTL is not None:
plt.subplot(sp_ls).set_axis_off()
plt.imshow(utils.t2np(iSL), cmap='gray')
plt.title('Source Label', font)
plt.subplot(sp_lt).set_axis_off()
plt.imshow(utils.t2np(iTL), cmap='gray')
plt.title('Target Label', font)
plt.subplot(sp_lw).set_axis_off()
plt.imshow(utils.t2np(iWL), cmap='gray')
plt.title('Warped Label', font)
if visual_param is not None:
if i == 0 and visual_param['visualize']:
plt.show()
if visual_param['save_fig']:
file_name = visual_param['pair_name'][i]
join_p = lambda pth1, pth2: os.path.join(pth1, pth2)
make_dir(join_p(visual_param['save_fig_path_byname'], file_name))
make_dir(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']))
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byname'], file_name),
visual_param['iter'] + extension),
dpi=dpi)
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']), file_name + extension),
dpi=dpi)
plt.clf()
else:
plt.show()
def _show_current_images_1d(iS,
iT,
iW,
iter,
vizImage,
vizName,
phiWarped,
visual_param=None,
i=0):
if (vizImage is not None) and (phiWarped is not None):
sp_s = 231
sp_t = 232
sp_w = 233
sp_p = 235
sp_v = 236
elif (vizImage is not None):
sp_s = 221
sp_t = 222
sp_w = 223
sp_v = 224
elif (phiWarped is not None):
sp_s = 221
sp_t = 222
sp_w = 223
sp_p = 224
else:
sp_s = 131
sp_t = 132
sp_w = 133
plt.suptitle('Iteration = ' + str(iter))
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
plt.subplot(sp_s)
plt.plot(utils.t2np(iS))
plt.title('source image')
plt.subplot(sp_t)
plt.plot(utils.t2np(iT))
plt.title('target image')
plt.subplot(sp_w)
plt.plot(utils.t2np(iT), 'g', linestyle=':')
plt.plot(utils.t2np(iS), 'r', linestyle='--')
plt.plot(utils.t2np(iW), linestyle=':')
plt.title('warped image')
if phiWarped is not None:
plt.subplot(sp_p)
plt.plot(utils.t2np(phiWarped[0, :]))
plt.title('phi')
if vizImage is not None:
plt.subplot(sp_v)
plt.plot(utils.lift_to_dimension(utils.t2np(vizImage), 1))
plt.title(vizName)
if visual_param is not None:
if i == 0 and visual_param['visualize']:
plt.show()
if visual_param['save_fig']:
file_name = visual_param['pair_name'][i]
join_p = lambda pth1, pth2: os.path.join(pth1, pth2)
make_dir(join_p(visual_param['save_fig_path_byname'], file_name))
make_dir(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']))
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byname'], file_name),
visual_param['iter'] + extension),
dpi=dpi)
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']), file_name + extension),
dpi=dpi)
else:
plt.show()
def _show_current_images_2d_no_map(iS,
iT,
iW,
iter,
vizImage,
vizName,
visual_param=None,
i=0):
if (vizImage is not None):
sp_s = 231
sp_t = 232
sp_w = 233
sp_c = 234
sp_v = 235
else:
sp_s = 221
sp_t = 222
sp_w = 223
sp_c = 224
plt.suptitle('Iteration = ' + str(iter))
plt.setp(plt.gcf(), 'facecolor', 'white')
plt.style.use('bmh')
plt.subplot(sp_s)
plt.imshow(utils.t2np(iS), cmap='gray')
plt.colorbar()
plt.title('source image')
plt.subplot(sp_t)
plt.imshow(utils.t2np(iT), cmap='gray')
plt.colorbar()
plt.title('target image')
plt.subplot(sp_w)
plt.imshow(utils.t2np(iW), cmap='gray')
plt.colorbar()
plt.title('warped image')
plt.subplot(sp_c)
plt.imshow(checkerboard_2d(utils.t2np(iW), utils.t2np(iT)), cmap='gray')
plt.colorbar()
plt.title('checkerboard')
if vizImage is not None:
plt.subplot(sp_v)
plt.imshow(utils.lift_to_dimension(utils.t2np(vizImage), 2),
cmap='gray')
plt.colorbar()
plt.title(vizName)
if visual_param is not None:
if i == 0 and visual_param['visualize']:
plt.show()
if visual_param['save_fig']:
file_name = visual_param['pair_name'][i]
join_p = lambda pth1, pth2: os.path.join(pth1, pth2)
make_dir(join_p(visual_param['save_fig_path_byname'], file_name))
make_dir(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']))
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byname'], file_name),
visual_param['iter'] + extension),
dpi=dpi)
plt.savefig(join_p(
join_p(visual_param['save_fig_path_byiter'],
visual_param['iter']), file_name + extension),
dpi=dpi)
else:
plt.show()