rishikksh20 / CrossViT-pytorch

Implementation of CrossViT: Cross-Attention Multi-Scale Vision Transformer for Image Classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multilabel Image classification

abhigoku10 opened this issue · comments

@rishikksh20 thanks for sharing code base , i had few queries
1.can we train crossvit for multilabel classification problem , if so what is the procedure
2. i have a custom dataset of 10.5k with 25class labels with instance as label vectors of 0 and 1
3. can remove the pre-trained classifer head and add our customr classifier ?

Thanks in advance