Hugging Face's logo

amaye15
/
SwinV2-Base-Document-Classifier

Image Classification
Transformers
Safetensors
swinv2
Generated from Trainer
Inference Endpoints
Model card Files Community
amaye15 commited on
Commit
b1d8947
•
1 Parent(s): 9678fba

Upload Swinv2ForImageClassification

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. config.json +3 -3
  3. model.safetensors +1 -1
README.md CHANGED
@@ -1,13 +1,13 @@
1
  ---
2
- license: apache-2.0
3
  base_model: microsoft/swinv2-base-patch4-window16-256
4
- tags:
5
- - generated_from_trainer
6
  metrics:
7
  - accuracy
8
  - f1
9
  - precision
10
  - recall
 
 
11
  model-index:
12
  - name: SwinV2-Base-Document-Classifier
13
  results: []
 
1
  ---
 
2
  base_model: microsoft/swinv2-base-patch4-window16-256
3
+ license: apache-2.0
 
4
  metrics:
5
  - accuracy
6
  - f1
7
  - precision
8
  - recall
9
+ tags:
10
+ - generated_from_trainer
11
  model-index:
12
  - name: SwinV2-Base-Document-Classifier
13
  results: []
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "microsoft/swinv2-base-patch4-window16-256",
3
  "architectures": [
4
  "Swinv2ForImageClassification"
5
  ],
@@ -20,8 +20,8 @@
20
  "0": "Barcode",
21
  "1": "Invoice",
22
  "2": "Object",
23
- "3": "Invoice",
24
- "4": "Barcode"
25
  },
26
  "image_size": 256,
27
  "initializer_range": 0.02,
 
1
  {
2
+ "_name_or_path": "/Users/andrewmayes/Project/document-type-detection/results/checkpoint-3000",
3
  "architectures": [
4
  "Swinv2ForImageClassification"
5
  ],
 
20
  "0": "Barcode",
21
  "1": "Invoice",
22
  "2": "Object",
23
+ "3": "Receipt",
24
+ "4": "Non-Object"
25
  },
26
  "image_size": 256,
27
  "initializer_range": 0.02,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:03bfc98ce4c44449680f0d5b59dce34e049d3041eea58ff47ae5179321a65bb8
3
  size 347657788
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e68df94c220d928bfdea9ab30da4b39ade950e39cb877e6198fdcff44599cf4
3
  size 347657788