utils

real_robot.utils.visualization.utils.colorize_mask(pred_mask: ndarray) → ndarray

Colorize a predicted mask :param pred_mask: [H, W] bool/np.uint8 np.ndarray :return mask: colorized mask, [H, W, 3] np.uint8 np.ndarray

real_robot.utils.visualization.utils.draw_mask(rgb_img, mask, alpha=0.5, id_countour=False) → ndarray

Overlay predicted mask on rgb image :param rgb_img: RGB image, [H, W, 3] np.uint8 np.ndarray :param mask: [H, W] bool/np.uint8 np.ndarray :param alpha: overlay transparency :return img_mask: mask-overlayed image, [H, W, 3] np.uint8 np.ndarray