/SupervisedSimSiam

Primary LanguageJupyter NotebookMIT LicenseMIT

SupervisedSimSiam

Task

Food101の分類

@TODO

[x] データローダの作成
[x] EfficientNetの実装
[x] SupervisedContrastiveLearningの実装
[ ] SimSiamによる学習の実装
[ ] MADGRAD Optimizerの実装

Setting

データセットの準備

Food101のデータセットのダウンロードと展開

./script/food101_dataset.sh

Ref

データセット

@inproceedings{bossard14,
  title = {Food-101 -- Mining Discriminative Components with Random Forests},
  author = {Bossard, Lukas and Guillaumin, Matthieu and Van Gool, Luc},
  booktitle = {European Conference on Computer Vision},
  year = {2014}
}

notebook

food101_effnet_train.ipynb