Update README.md
Browse files
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 |
-
|
|
| 62 |
-
|-------
|
| 63 |
-
|
|
| 64 |
-
| vit_large_patch32_224
|
| 65 |
-
| vit_tiny_patch16_224
|
| 66 |
-
| vit_small_patch32_224
|
| 67 |
-
| vit_small_patch16_224
|
| 68 |
-
| vit_base_patch16_224
|
| 69 |
-
| vit_base_patch32_224
|
|
|
|
|
|
|
|
|
|
| 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 |
|