This is the code used to train and test the models used in "Comparative Study Between Vision Transformer and EfficientNet on Marsh Grass Classification".
This paper was accepted to FLAIRS 2023 for the main conference track
install additional dependencies:
pip install efficientnet_pytorch
pip install pytorch_pretrained_vit
Use main.py to run code