wbw520 / MTUNet

MTUNet: Few-shot Image Classification with Visual Explanations (CVPRW 2021)
29 stars 3 forks source link

Add some of my own dataset #3

Open Troyella opened 2 years ago

Troyella commented 2 years ago

Hello! I want to add some new dataset into cifarfs and train, where is the parameters should I change? I have already put the dataset into cifarfs and generate the cifar100 .

wbw520 commented 2 years ago

The split of cifarfs is based on the code from https://github.com/bertinetto/r2d2. It uses the CSV file which is also utilized in the data loader for my code. If you add your own dataset, the CSV file should also be modified.

Troyella commented 2 years ago

OK, thank you very much.