multi class question - Githubissues

Karel911 / TRACER

TRACER: Extreme Attention Guided Salient Object Tracing Network (AAAI 2022) implementation in PyTorch

Apache License 2.0

195 stars 41 forks source link

Open DeepHM opened 2 years ago

DeepHM commented 2 years ago

Hello. Thank you for your wonderful research.

I want to extend your methodology to multi-class classification. Is this possible? If possible, could you give me some implementation tips?

Karel911 commented 2 years ago

The multi-class classification you mean is such a semantic segmentation task?

DeepHM commented 2 years ago

Yes! I would like to apply your methodology to a segmentation task with 3 or 5 classes.

DAgalaxy commented 2 years ago

Hello, I also would like to apply your API_loss to my 3_classes semantic segmentation task, but i am not sure it will be work or not?

Karel911 commented 2 years ago

Hi @DeepHM, Sorry for the late reply. Does your network return the shape of X \in R^{B, n_classes, H, W}?