Skip to content

Keras/Tensorflow implementation for co-generation and segmentation of surgical instruments using unlabelled robot-assisted surgery data.

License

Notifications You must be signed in to change notification settings

tajwarabraraleef/coSegGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Co-Generation and Segmentation for Generalized Surgical Instrument Segmentation on Unlabelled Data

Keras/Tensorflow implementation of coSegGAN. This work has been published in Medical Image Computing and Computer Assisted Intervention – MICCAI 2021 (https://miccai2021.org/en/).

[ arXiv / Paper ]

Dependencies

Python 3.6
Tensorflow: 2.0.0
Keras: 2.3.1

Environment Setup

Recreate conda environment as follows:

conda env create -f environment.yml

Or if you are using Docker:

docker pull tazleef/tf2.0.0-cv-keras2.3.1-imgaug:latest

Dataset

Links to the publicly available dataset used in this work:

Training

Due to privacy policy, we are unable to share our surgical dataset. However, we intend to release them soon once we get the approval. For now, we have included a few sample cases for reference. Alternatively, if you have your own data, format it in the same way and set the filepath and training parameters in data_path_Loader.py and train.py.

To train the model, run train.py.

This code can also be used for other segmentation tasks with domain variations.

Citation

If you use this code or the provided model, please cite the following paper:

@inproceedings{kalia2021co,
  title={Co-Generation and Segmentation for Generalized Surgical Instrument Segmentation on Unlabelled Data},
  author={Kalia, Megha and Aleef, Tajwar Abrar and Navab, Nassir and Black, Peter and Salcudean, Septimiu E},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={403--412},
  year={2021},
  organization={Springer}
}
}

Contact

If you face any problem using this code then create an issue in this repository or contact me at tajwaraleef@ece.ubc.ca

Acknowledgements

The code is based on https://github.com/eriklindernoren/Keras-GAN/tree/master/cyclegan

License

MIT

About

Keras/Tensorflow implementation for co-generation and segmentation of surgical instruments using unlabelled robot-assisted surgery data.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages