Mask R-CNN is a convolution based network for object instance segmentation. This implementation provides 1.3x faster training while maintaining target accuracy.
Mask R-CNN builds on top of Faster R-CNN adding a mask head for the task of image segmentation.
The architecture consists of the following:
Figure 1. Diagram of Mask R-CNN framework from original paper
This model was trained using script available on NGC and in GitHub repo.
The following datasets were used to train this model:
Performance numbers for this model are available in NGC.