How train with a new dataset
mohammad-saber opened this issue · 1 comments
mohammad-saber commented
Thank you very much for sharing your code.
I want to train models with a dataset other than miniImageNet or Omniglot. Could you guide me how to arrange dataset (main folder , sub-folder), and how to train models on a new dataset?
Thanks a lot.
yinboc commented
Hi. The format of this miniImageNet is that, there are .csv files which specify the image list in each split, and an "image" folder that contains all the images, you may have a look at the "materials" folder after downloading the dataset.
Besides following the original format, you can also write a Pytorch dataset class for replacing mini_imagenet.py.