This repository hosts the MushR Mask R CNN model for maturity detection of Oyster mushrooms.
To run inference, please follow the official instructions to install detectron2.
All training related configuration can be found in train_2023_04_04/.
We used the train.sh script to train the model.
To continue training from the last checkpoint, use: continue_training.sh
The model weights of our trained model can be found in https://github.com/ETCE-LAB/mushr-mask-r-cnn/releases
The scripts, inference_images.sh and inference_images-single.sh can be used to run inference.
Please check inference_2023_04_20 for examples of expected results.
To evaluate the model, please use the eval.sh script.
Our evaluation results of the trained model are given in eval_2023_07_19.