Update README.md
Browse files
README.md
CHANGED
|
@@ -15,11 +15,7 @@ datasets:
|
|
| 15 |
metrics:
|
| 16 |
- accuracy
|
| 17 |
pipeline_tag: image-classification
|
| 18 |
-
|
| 19 |
-
- src: https://huggingface.co/datasets/rvl_cdip/resolve/main/sample_images/letter_0.jpg
|
| 20 |
-
example_title: Letter
|
| 21 |
-
- src: https://huggingface.co/datasets/rvl_cdip/resolve/main/sample_images/form_0.jpg
|
| 22 |
-
example_title: Form
|
| 23 |
---
|
| 24 |
|
| 25 |
# ViT-Tiny Classifier for RVL-CDIP Document Classification (Distilled)
|
|
@@ -92,7 +88,7 @@ print("Predicted class:", predicted_class)
|
|
| 92 |
|
| 93 |
| Metric | Value |
|
| 94 |
|--------|-------|
|
| 95 |
-
| Accuracy |
|
| 96 |
| Parameters | ~5.5M |
|
| 97 |
| Model Size | ~22 MB |
|
| 98 |
| Input Size | 224x224 pixels |
|
|
|
|
| 15 |
metrics:
|
| 16 |
- accuracy
|
| 17 |
pipeline_tag: image-classification
|
| 18 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
---
|
| 20 |
|
| 21 |
# ViT-Tiny Classifier for RVL-CDIP Document Classification (Distilled)
|
|
|
|
| 88 |
|
| 89 |
| Metric | Value |
|
| 90 |
|--------|-------|
|
| 91 |
+
| Accuracy | 0.9210 |
|
| 92 |
| Parameters | ~5.5M |
|
| 93 |
| Model Size | ~22 MB |
|
| 94 |
| Input Size | 224x224 pixels |
|