Can I predict multiclass object from this code?

JunMa11 / SegWithDistMap

How Distance Transform Maps Boost Segmentation CNNs: An Empirical Study

https://openreview.net/forum?id=hM4pNbXWst

Apache License 2.0

375 stars 67 forks source link

Closed Hung-Hsiang-617 closed 3 years ago

Hung-Hsiang-617 commented 3 years ago

How to predict multiple categories? Does the loss function support multiple categories? Thany you.

JunMa11 commented 3 years ago

Hi @Hung-Hsiang-617 ,

Sorry for my late reply.

The multi-class version is available at this repo https://github.com/JunMa11/SegLoss