search

kaixin96 / PANet

Code for our ICCV 2019 paper PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment
315 stars 64 forks source link

How can I train it on my data? #39

Open shankrmahadevan opened 3 years ago

shankrmahadevan commented 3 years ago

Sir, I'd want to use the PANet to do few-shot segmentation on a dataset I've gathered. I looked over the code but couldn't figure out how to train/test the model with an external dataset. Could you please include a script for this?

kaixin96 commented 3 years ago

Hi @shankrmahadevan , the data loading code is quite VOC/COCO specific. If you want to use an external dataset, I suggest you checking out torchmeta. It has a cleaner abstraction for dataset class tailored for few-shot tasks. Thank you.

Meoooww commented 2 years ago

Did you successfully apply it?