This project was created based on implementation of multiple granularities resnet link that was modified for this task. This approach is based on this paper.
Pretrained model can be downloaded here
python main.py --data_path "folder0" "folder2" --weight "./model.pt" --output_file "result.csv"