Create your list of filenames and their corresponding segmentation maps with generate_ann.py
.
The models can be trained by running train.py
. Check the argument_parser.py to ensure a good configuration of running. Also, set up your root directory in train.py
.
For saving the results, you can make use of save_results.py
.
A DeepLab-based pre-trained model with DRN backbone can be found here. This can be used to get some preliminary results of the tools and colon folds detection. We recommend using the Dexined for the colon folds prediction instead of this model. However, for tools segmentation you may prefer a model similar to this one.