Check out the Hugging Face repository to view and download the model: https://huggingface.co/SABR22/ViT-threat-classification-v2
This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an amalgamated dataset open images. This is model created as a prrof of concept for a Carleton University computer vision event. It is by no means meant to be used in deliverable systems in its current state, and should be used exclusively for research and development.
It achieves the following results on the evaluation set:
- Loss: 0.0381
- F1: 0.9657
- Precision: 0.9563
- Recall: 0.9752