Giters

oscarknagg / few-shot

Repository for few-shot learning machine learning projects

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Wrong number of classes when rotating

Franklin-Yao opened this issue · comments

commented

There is a bug related to class names. Using your code in scripts/prepare_omniglot.py, I got class names like 'Alphabet_of_the_Magi.0.character03'. But I think 'Alphabet_of_the_Magi.0.character03' and 'Alphabet_of_the_Magi.90.character03' have same classes because they are augmented data which have same labels(classes).
So if you change the following line
https://github.com/oscarknagg/few-shot/blob/master/few_shot/datasets.py#L80
to
alphabet = root.split('/')[-2].split('.')[0]