Can dice loss be used for multi-class or rewriten for multi-class? - Githubissues

mattmacy / vnet.pytorch

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

https://mattmacy.github.io/vnet.pytorch

BSD 3-Clause "New" or "Revised" License

693 stars 200 forks source link

Can dice loss be used for multi-class or rewriten for multi-class? #9

Open elias-1 opened 7 years ago

elias-1 commented 7 years ago

Can dice loss be used for multi-class or rewriten for multi-class?

hfarhidzadeh commented 6 years ago

Yes, it is easy to manipulate it to work for mutli-class problem. Check http://www.niftynet.io/. Could you use dice loss on multi-Gpu?