nabsabraham / focal-tversky-unet

This repo contains the code for our paper "A novel focal Tversky loss function and improved Attention U-Net for lesion segmentation" accepted at IEEE ISBI 2019.
357 stars 72 forks source link

Multi class support #20

Open riyaj8888 opened 4 years ago

riyaj8888 commented 4 years ago

Can we use the tversky loss for multiple classes ? Or do we have to make changes accordingly.

Zhiyu-Xu commented 4 years ago

Can we use the tversky loss for multiple classes ? Or do we have to make changes accordingly.

Hello, have you solved this problem?

kaczmarj commented 2 years ago

see https://github.com/nabsabraham/focal-tversky-unet/issues/3