search

QData / C-Tran

General Multi-label Image Classification with Transformers
MIT License
238 stars 40 forks source link

Multi-Label Classification where each label is Multi-class #22

Open NarenZen opened 1 year ago

NarenZen commented 1 year ago

how to train C-Tran for Multi-class possibilities.

What I mean is pug, husky, dolmination (outputs one label out of 3 possibilities) cat/ not a cat (outputs one label out of 2 possibilities)