This model is a fine-tuned version of
google/vit-base-patch16-224-in21k on a threat classification dataset.
This model was created for a Carleton University computer vision hacking event and serves as a proof of concept rather than complete model. It is trained on an extremely small and limited dataset and the performance is limited.
It achieves the following results on the evaluation set:
