sdcjimmy / Mask-RCNN

A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Mask-RCNN

A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch

  1. model.py includes the models of ResNet and FPN which were already implemented by the authors of the papers and reproduced in this implementation
  2. nms and RoiAlign are taken from Robb Girshick's implementation of faster RCNN
  3. Focal loss has been added to this implementtaion on lieu of better results as evidenced by the paper on RetinaNets

About

A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch

License:Other


Languages

Language:Python 83.9%Language:C 10.0%Language:Cuda 6.0%Language:C++ 0.1%