mask_rcnn_extensions

Description

Custom modules for handling and modifying the pytorch Mask R-CNN model.

Classes

MaskRCNNBoxToSegMask	Turn batch bounding box output of torchvision Mask R-CNN into segmentation masks for given label.
MaskRCNNToSegMask	Turn batch segmentation output of torchvision Mask R-CNN into segmentation masks for given label.