juhongm999 / hsnet

Official PyTorch Implementation of Hypercorrelation Squeeze for Few-Shot Segmentation, ICCV 2021
231 stars 43 forks source link

train with my data #33

Open hadi1994shokati opened 11 months ago

hadi1994shokati commented 11 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.