pranav2711 commited on
Commit
d54ebb8
·
verified ·
1 Parent(s): e4070eb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -11
README.md CHANGED
@@ -56,17 +56,20 @@ They are well-suited for deployment in digital forensics, content moderation, an
56
 
57
  ---
58
 
59
- ## 📊 Results
60
-
61
- | Vision Transformer Model | Train Accuracy | Validation Accuracy |
62
- |-------------------------------|----------------|---------------------|
63
- | **vit_large_patch16_224** | 94.89% | **91.22%** |
64
- | vit_large_patch32_224 | 91.31% | 89.23% |
65
- | vit_tiny_patch16_224 | 92.41% | 89.20% |
66
- | vit_small_patch32_224 | 91.58% | 88.38% |
67
- | vit_small_patch16_224 | 84.72% | 87.68% |
68
- | vit_base_patch16_224 | 90.65% | 85.36% |
69
- | vit_base_patch32_224 | 79.54% | 79.54% |
 
 
 
70
 
71
  ---
72
 
 
56
 
57
  ---
58
 
59
+ ## 📊 Results
60
+
61
+ | Model | Train Accuracy | Validation Accuracy |
62
+ |-------|----------------|---------------------|
63
+ | [vit_large_patch16_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_large_patch16_224.pth) | **94.89%** | **91.22%** |
64
+ | [vit_large_patch32_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_large_patch32_224.pth) | 91.31% | 89.23% |
65
+ | [vit_tiny_patch16_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_tiny_patch16_224.pth) | 92.41% | 89.20% |
66
+ | [vit_small_patch32_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_small_patch32_224.pth) | 91.38% | 88.29% |
67
+ | [vit_small_patch16_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_small_patch32_224.pth) | 80.67% | 81.25% |
68
+ | [vit_base_patch16_224](http://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_base_patch16_224.pth) | 90.65% | 85.36% |
69
+ | [vit_base_patch32_224](https://huggingface.co/pranav2711/VisionTransformerDigitalForensics/blob/main/vit_base_patch32_224.pth) | 79.54% | 79.54% |
70
+
71
+ **vit_large_patch16_224**
72
+ This model achieved the highest validation accuracy of 91.22% with strong training stability and generalization. It is recommended as the final model for deployment or downstream tasks.
73
 
74
  ---
75