/siege_mask_rnn

custom mask r-cnn model

Primary LanguageJupyter Notebook

siege_mask_rnn

Link to download the dataset here

Trained a mask R-CNN model on a custom rainbow six siege dataset. For more details check out my medium posts https://medium.com/@michaelsugimura

This repo is mostly a modification on the matterport mask r-cnn repo

I was not able to include the model files or image dataset, I would be happy to provide them if anyone ends up being interested.

Got around to uploading them somewhere so here is a link to the dataset stored on google Drive.

-siege_output folder contains some sample images that I passed through the model to see how it was performing

-mrcnn contains the util and model files for running the network. These are from matterport

-the siege_m_rcnn.py file is my modification of their balloon sample

-the inspection file is also a modified version of the balloon sample