Simple Mask RCNN implementation in Pytorch for educational purposes
- Load image & Normalize with pytorch Transform
- Implement config with yacs
- Create MaskRCNN class with extractor from torchvision.models
- Implement RPN module
- Implement ROI Pooling module
- Load pretrained model