Downloads · 30 days
4
11% of all-time downloads
Aby101/visual-vit
visual-vit is a machine learning model from Aby101. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
4
11% of all-time downloads
All-time downloads
35
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
use link
Deepfake Image Detection Using Fine-Tuned Vision Transformer (ViT) This project focuses on detecting deepfake images using a fine-tuned version of the pre-trained model google/vit-base-patch16-224-in21k. The approach leverages the power of Vision Transformers (ViT) to classify images as real or fake.
Model Overview Base Model: google/vit-base-patch16-224-in21k Dataset: deepfake and real images .https://www.kaggle.com/datasets/manjilkarki/deepfake-and-real-images Classes: Binary classification (Fake, Real) Performance: Validation Accuracy: 97% Test Accuracy: 92% Figure 1: Confusion matrix for test data