shailesharya/Segmentation

Train Mask R-CNN for Image Segmentation

Segmentation

Steps:

1. Annotating Dataset

• Dataset
• Annotation tool

2. Training model:

• Mask-RCNN
• Trained on custom dataset (Street signs and Ad boards)

2.1 Code Structure:

• Segmentation.ipynb: Code for training Mask-RCNN with custom dataset
• Segmentation_Test.ipynb: Code for running inference on trained model

3. Result:

• output.avi