Implementation of Mask R-CNN in PyTorch

Hi there!

Does anybody have implementation of Mask R-CNN in PyTorch that has ability to fine-tuning on own dataset?

1 Like

Hi, Oktai,


1 Like

@zhanghaoinf thanks a lot!

Did you try to train on own dataset with second model?

Not yet, I just saw a guy I stared follow the second implementation.

The author of the second implementation posts benchmark evaluation results, seems reasonable.

From the Facebook Research team, just released, support for training on own dataset.

1 Like