CE-Net

The manuscript has been accepted in TMI.

Please start up the "visdom" before running the main.py. Then, run the main.py file.

We have uploaded the DRIVE dataset to run the retinal vessel detection. The other medical datasets will be uploaded in the next submission.

The submission mainly contains:

1. architecture (called CE-Net) in networks/cenet.py
2. multi-class dice loss in loss.py
3. data augmentation in data.py