Where is a training code to train GeM baseline? https://github.com/facebookresearch/isc2021/blob/master/baselines/GeM_baseline.py
Hi @LexiBender, welcome to the competition. The script you linked is indeed the code for the GeM (a.k.a. MultGrain) baseline model. That model uses a pretrained model (so no training in that script) and a PCA stage that is trained on the training images.
The repo for the baseline models has some walkthroughs for using the GeM/MultiGrain model. If you’re looking for more details about the pretrained MultiGrain model, that first walkthrough page (#4) includes some links to the original MultiGrain paper and its accompanying GitHub repo.
- isc2021/4_multigrain_subset.md at master · facebookresearch/isc2021 · GitHub
- isc2021/5_multigrain_full.md at master · facebookresearch/isc2021 · GitHub
Hope that helps!
Thank you for the response, got it!
thanks for the awesome information.