search

matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Other
24.57k stars 11.69k forks source link

Mask_RCNN, Backbone modification #1965

Open amuthalingeswaranbose opened 4 years ago

amuthalingeswaranbose commented 4 years ago

I have been using the Mask_RCNN algorithm. It works fine. I have followed the process of training and validation followed by testing. But now, I would like to change the Backbone, ResNet101 to a new one. Is that feasible? What are the other changes that i would have to bring in the code, and the parameters. Kindly help.

truongtd6285 commented 4 years ago

You have two options: BACKBONE = "resnet50" or BACKBONE = "resnet101". You can change in the config.py file. See here: https://github.com/matterport/Mask_RCNN/wiki