food-recognition

Brief Description

Repo with some work done on using Swin Transformer on the Food2k Dataset

Folders

• food-recognition: Python package containing all needed functions to train and extracting class attention maps on Swin Transformer

• tutorial: Simple notebooks that show how to train and extract class activation map from pretrained Swin transformer

References

• Explainability for Vision Transformers (in PyTorch)
• Code release for Large Scale Visual Food Recognition
• Transformer-Explainability