Python 2.7 -- Cuda-8.0 -- TensorFlow 1.4.0 -- Keras 2.1.5
This project is inspired in the 3 papers below with implementation in Keras, used in a 3D dataset. The application is the segmentation of the cerebrovascular system in MRA phase contrastusing images.
- SegAN: Adversarial Network with Multi-scale L1 Loss for Medical Image Segmentation by Yuan Xue, Tao Xu, Han Zhang, L. Rodney Long, Xiaolei Huang.
- Adversarial Learning with Multi-Scale Loss for Skin Lesion Segmentation by Yuan Xue, Tao Xu, Xiaolei Huang.
- Semantic Segmentation using Adversarial Networks by Pauline Luc, Camille Couprie, Soumith Chintala, Jakob Verbeek.
In the segmentor part of the network, I am also using a dice score as a mixed loss function.