search

juhongm999 / hsnet

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

Train/Test on Own data #10

Closed shankrmahadevan closed 3 years ago

shankrmahadevan commented 3 years ago

Hi Sir, how can I train/test this model on my own dataset?

juhongm999 commented 3 years ago

You have to manually implement your Dataset and Dataloader. See https://pytorch.org/tutorials/beginner/data_loading_tutorial.html

shankrmahadevan commented 3 years ago

Thank you, Sir!