zzdxjtu / CE-Net

The manuscript has been accepted in TMI.

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

Update: We have modified the loss function. The cuda error (or warning) will not occur.