Multiple classes training
scpctshp opened this issue · comments
scpctshp commented
Hi, is it possible to use this model for multi-classes segmentation training? Thanks.
Taehun Kim commented
Hello, sorry for the very late reply.
Yes, but you need some modifications. You need to change the output class number from 1 to the desired number, change the loss function for multi-class, change dataloader and custom transform functions.
Taehun Kim commented
Also, you can find similar issue from PraNet #9.