facebookresearch/maskrcnn-benchmark

This project aims at providing the necessary building blocks for easily creating detection and segmentation models using PyTorch 1.0. A notebook with the demo can be found in demo/Mask_R-CNN_demo.ipynb. Check INSTALL.md for installation instructions. Pre-trained models, baselines and comparison with Detectron and mmdetection can be found in MODEL_ZOO.md We provide a helper class to simplify writing inference pipelines using pre-trained models.