Official implementation of RA-L 2021 paper Multi-frame Feature Aggregation for Real-time Instrument Segmentation in Endoscopic Video.
- Creat the conda environment with required packages using the following command:
conda env create -f resources/env.yml
- You could download the dataset from here.
- Update the datase directory in main.sh.
- Train using the following command:
./train.sh
- Test using the following command:
./test.sh
You could download the checkpoint from here.