How to train the multi-class task

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

2.66k stars 594 forks source link

Closed CC-Xiaoyu closed 1 year ago

CC-Xiaoyu commented 2 years ago

Hello,i want to train a multi-class task,What needs to be changed,such as Loss and so on Thank you~