Requirements:
- Mask R-CNN(Clone the repo)
- mask_rcnn_coco.h5
Train a new model starting from pre-trained COCO weights
python3 custom.py train --dataset=/path/to/datasetfolder --weights=coco
Use inspect_custom_model.ipynb to check custom images
Requirements:
Train a new model starting from pre-trained COCO weights
python3 custom.py train --dataset=/path/to/datasetfolder --weights=coco
Use inspect_custom_model.ipynb to check custom images