zokyo.utils.semantic_seg module

zokyo.utils.semantic_seg.generate_semantic_map(img_path, ann_path, out_dir=None, opacity=0.4, gt=True, color_map=None, fill_poly=False, verbose=True)

Generate semantic segmentation map for the given image and its annotations and save it to given output directory if given. If gt is true then ground truth mask is also returned. If color map list is not given then Spectral color map from matplotlib is used. (Currently accepts coco toyo annotations)