Architecture of Mask RCNN – A PyTorch Implementation
github.com
github.com
yeah dumping it all in a single 2098 line file, why not? Here is a clean documented implementation of faster rcnn for comparison: https://github.com/chenyuntc/simple-faster-rcnn-pytorch
Academics are great at finding "local optima," but rarely do they see beyond their next result, and then others have the onerous task of trying to verify the previous results or worse still using the previous work to build towards "larger" results.
From what I recall about Faster R-CNN, the Regions Of Interest (ROI) are pre-determined via Selective Search, right? So I presume you would need to do the same thing with Mask-RCNN? I think this is the part I am the most confused with since I have never implemented Selective Search myself. Could you point me to introductory material on it?
Lastly, I can see the author of this work has read my blog post on understanding SSD MultiBox - glad it helped in some way :).
Selective search is implemented in Fast-RCNN. Faster-RCNN improves upon that and uses a Region Proposal Me to propose RoI that may contain objects which speed up training and inference time.