Commit ·
c1f238c
1
Parent(s): d26b16c
Update README.md
Browse files
README.md
CHANGED
|
@@ -6,13 +6,15 @@ tags:
|
|
| 6 |
metrics:
|
| 7 |
- f1
|
| 8 |
model-index:
|
| 9 |
-
- name:
|
| 10 |
results: []
|
| 11 |
---
|
| 12 |
|
| 13 |
-
#
|
| 14 |
|
| 15 |
-
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the cord, rvl-cdip, visual-genome and an external receipt
|
|
|
|
|
|
|
| 16 |
|
| 17 |
It achieves the following results on the evaluation set, which contain pictures from the above datasets in scanned, photography or mobile picture formats (color and grayscale):
|
| 18 |
- Loss: 0.0116
|
|
@@ -20,7 +22,7 @@ It achieves the following results on the evaluation set, which contain pictures
|
|
| 20 |
|
| 21 |
## Model description
|
| 22 |
|
| 23 |
-
This model is a Binary Classifier finetuned version of ViT, to predict if an input image is a picture / scan of
|
| 24 |
|
| 25 |
## Intended uses & limitations
|
| 26 |
|
|
|
|
| 6 |
metrics:
|
| 7 |
- f1
|
| 8 |
model-index:
|
| 9 |
+
- name: vit_receipts_classifier
|
| 10 |
results: []
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# vit_receipts_classifier
|
| 14 |
|
| 15 |
+
This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the cord, rvl-cdip, visual-genome and an external receipt dataset to carry out Binary Classification (`ticket` vs `no_ticket`).
|
| 16 |
+
|
| 17 |
+
Ticket here is used as a synonym to "receipt".
|
| 18 |
|
| 19 |
It achieves the following results on the evaluation set, which contain pictures from the above datasets in scanned, photography or mobile picture formats (color and grayscale):
|
| 20 |
- Loss: 0.0116
|
|
|
|
| 22 |
|
| 23 |
## Model description
|
| 24 |
|
| 25 |
+
This model is a Binary Classifier finetuned version of ViT, to predict if an input image is a picture / scan of receipts(s) o something else.
|
| 26 |
|
| 27 |
## Intended uses & limitations
|
| 28 |
|