epic_kitchens.masks.visualisation

Module Contents

Classes

DetectionRenderer

Functions

resize_mask(mask: np.ndarray, height: int, width: int, smooth: bool = True) → np.ndarray
resize_bbox(bbox: BBox, height: int, width: int) → Tuple[int, int, int, int]

epic_kitchens.masks.visualisation.resize_mask(mask: np.ndarray, height: int, width: int, smooth: bool = True) → np.ndarray

epic_kitchens.masks.visualisation.resize_bbox(bbox: BBox, height: int, width: int) → Tuple[int, int, int, int]

class epic_kitchens.masks.visualisation.DetectionRenderer(display_mask: bool = True, display_bbox: bool = True, score_threshold: float = 0, smooth_mask: bool = True)

render_detections(self, img: PIL.Image.Image, detection: FrameObjectDetections)