Mask R-CNN
Folder containing code to train and produce inferences using a Mask R-CNN model for the Breast Cancer tumor detection and classification task.
In order to run the code, the user must supply the image datasets directory as a zip file. The image datasets can be found here: https://www.kaggle.com/datasets/aryashah2k/breast-ultrasound-images-dataset/code
Use the Jupyter notebook file mask_rcnn.ipynb to run the scripts. Additional files contain helper functions that are imported in the Jupyter notebook.
BUSI.ipynb
Contains the code to train and run inference on the U-Net model.
First_Pytorch.ipynb
Contains the code to train nd run inference on the ResNet18 model.