AIVResearch / MSANet

Official Pytorch implementation of Multi-Similarity and Attention Guidence for Boosting Few-Shot Segmentation.
https://arxiv.org/abs/2206.09667v1
63 stars 12 forks source link

Train on myown data #13

Open hadi1994shokati opened 10 months ago

hadi1994shokati commented 10 months ago

Hello. Thanks for sharing the code. I want to train the model with my data. Instead of 20 Pascal classes, my data has only 1 class and it is binary classification. Should I just use a split? What parts of the code should I change? Looking forward to hearing you.