def _visual_image_ref_warp_lnds_move_warp(record, path_dataset=None,
path_experiment=None):
""" visualise the case with warped reference landmarks to the move frame
:param {} record: row with the experiment
:param str|None path_dataset: path to the dataset folder
:param str|None path_experiment: path to the experiment folder
:return obj|None:
"""
assert COL_POINTS_REF_WARP in record and isinstance(record[COL_POINTS_REF_WARP], str), \
'Missing registered image "%s"' % COL_POINTS_REF_WARP
path_points_warp = update_path_(record[COL_POINTS_REF_WARP], path_experiment)
if not os.path.isfile(path_points_warp):
logging.warning('missing warped landmarks for: %r', dict(record))
return
points_ref, points_move, path_img_ref = _load_landmarks(record, path_dataset)
points_warp = load_landmarks(path_points_warp)
if not list(points_warp):
return
# draw image with landmarks
image_move = load_image(update_path_(record[COL_IMAGE_MOVE], path_dataset))
# image_warp = tl_io.load_image(row['Moving image, Transf.'])
image = draw_image_points(image_move, points_warp)
save_image(os.path.join(update_path_(record[COL_REG_DIR], path_experiment),
NAME_IMAGE_REF_POINTS_WARP), image)
del image
# visualise the landmarks move during registration
image_ref = load_image(path_img_ref)
fig = draw_images_warped_landmarks(image_ref, image_move, points_ref,
points_move, points_warp)
del image_ref, image_move
return fig
def _visual_image_move_warp_lnds_ref_warp(cls,
item,
path_dataset=None,
path_experiment=None):
""" visualise the case with warped reference landmarks to the move frame
:param dict item: row with the experiment
:param str|None path_dataset: path to the dataset folder
:param str|None path_experiment: path to the experiment folder
:return obj|None:
"""
assert isinstance(item.get(cls.COL_POINTS_REF_WARP, None), str), \
'Missing registered points in "%s"' % cls.COL_POINTS_REF_WARP
path_points_warp = update_path(item[cls.COL_POINTS_REF_WARP],
pre_path=path_experiment)
if not os.path.isfile(path_points_warp):
logging.warning('missing warped landmarks for: %r', dict(item))
return
points_ref, points_move, path_img_ref = cls._load_landmarks(
item, path_dataset)
points_warp = load_landmarks(path_points_warp)
if not list(points_warp):
return
# draw image with landmarks
image_move = load_image(
update_path(item[cls.COL_IMAGE_MOVE], pre_path=path_dataset))
image = draw_image_points(image_move, points_warp)
save_image(
os.path.join(
update_path(item[cls.COL_REG_DIR], pre_path=path_experiment),
cls.NAME_IMAGE_REF_POINTS_WARP), image)
del image
image_ref = load_image(path_img_ref)
image_warp = cls._load_warped_image(item, path_experiment)
image = overlap_two_images(image_ref, image_warp)
save_image(
os.path.join(
update_path(item[cls.COL_REG_DIR], pre_path=path_experiment),
cls.NAME_IMAGE_REF_WARP), image)
del image, image_warp
# visualise the landmarks move during registration
fig = draw_images_warped_landmarks(image_ref, image_move, points_ref,
points_move, points_warp)
del image_ref, image_move
return fig