phunc20 commited on
Commit
7c20624
·
verified ·
1 Parent(s): e398e34

Update README.md

Browse files

1st edit on model card

Files changed (1) hide show
  1. README.md +38 -122
README.md CHANGED
@@ -1,11 +1,15 @@
1
  ---
2
  library_name: transformers
3
- tags: []
 
 
 
 
4
  ---
5
 
6
- # Model Card for Model ID
7
 
8
- <!-- Provide a quick summary of what the model is/does. -->
9
 
10
 
11
 
@@ -13,53 +17,26 @@ tags: []
13
 
14
  ### Model Description
15
 
16
- <!-- Provide a longer summary of what this model is. -->
17
-
18
- This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
19
-
20
- - **Developed by:** [More Information Needed]
21
- - **Funded by [optional]:** [More Information Needed]
22
- - **Shared by [optional]:** [More Information Needed]
23
- - **Model type:** [More Information Needed]
24
- - **Language(s) (NLP):** [More Information Needed]
25
- - **License:** [More Information Needed]
26
- - **Finetuned from model [optional]:** [More Information Needed]
27
-
28
- ### Model Sources [optional]
29
-
30
- <!-- Provide the basic links for the model. -->
31
-
32
- - **Repository:** [More Information Needed]
33
- - **Paper [optional]:** [More Information Needed]
34
- - **Demo [optional]:** [More Information Needed]
35
 
36
  ## Uses
37
 
38
  <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
 
 
40
  ### Direct Use
41
 
42
  <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
 
44
  [More Information Needed]
45
 
46
- ### Downstream Use [optional]
47
-
48
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
-
50
- [More Information Needed]
51
-
52
- ### Out-of-Scope Use
53
-
54
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
-
56
- [More Information Needed]
57
 
58
  ## Bias, Risks, and Limitations
59
 
60
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
-
62
- [More Information Needed]
63
 
64
  ### Recommendations
65
 
@@ -77,40 +54,26 @@ Use the code below to get started with the model.
77
 
78
  ### Training Data
79
 
80
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
-
82
- [More Information Needed]
83
 
84
  ### Training Procedure
85
 
86
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
-
88
- #### Preprocessing [optional]
89
-
90
- [More Information Needed]
91
-
92
-
93
- #### Training Hyperparameters
94
-
95
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
-
97
- #### Speeds, Sizes, Times [optional]
98
-
99
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
-
101
- [More Information Needed]
102
 
103
  ## Evaluation
104
 
105
- <!-- This section describes the evaluation protocols and provides the results. -->
106
-
107
  ### Testing Data, Factors & Metrics
108
 
109
  #### Testing Data
110
 
111
- <!-- This should link to a Dataset Card if possible. -->
112
-
113
- [More Information Needed]
114
 
115
  #### Factors
116
 
@@ -120,80 +83,33 @@ Use the code below to get started with the model.
120
 
121
  #### Metrics
122
 
123
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
-
125
- [More Information Needed]
 
126
 
127
  ### Results
 
128
 
129
- [More Information Needed]
130
-
131
- #### Summary
 
132
 
 
133
 
 
 
 
 
134
 
135
- ## Model Examination [optional]
136
-
137
- <!-- Relevant interpretability work for the model goes here -->
138
-
139
- [More Information Needed]
140
 
141
  ## Environmental Impact
142
 
143
- <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
-
145
  Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
 
147
  - **Hardware Type:** [More Information Needed]
148
  - **Hours used:** [More Information Needed]
149
  - **Cloud Provider:** [More Information Needed]
150
  - **Compute Region:** [More Information Needed]
151
- - **Carbon Emitted:** [More Information Needed]
152
-
153
- ## Technical Specifications [optional]
154
-
155
- ### Model Architecture and Objective
156
-
157
- [More Information Needed]
158
-
159
- ### Compute Infrastructure
160
-
161
- [More Information Needed]
162
-
163
- #### Hardware
164
-
165
- [More Information Needed]
166
-
167
- #### Software
168
-
169
- [More Information Needed]
170
-
171
- ## Citation [optional]
172
-
173
- <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
-
175
- **BibTeX:**
176
-
177
- [More Information Needed]
178
-
179
- **APA:**
180
-
181
- [More Information Needed]
182
-
183
- ## Glossary [optional]
184
-
185
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
-
187
- [More Information Needed]
188
-
189
- ## More Information [optional]
190
-
191
- [More Information Needed]
192
-
193
- ## Model Card Authors [optional]
194
-
195
- [More Information Needed]
196
-
197
- ## Model Card Contact
198
-
199
- [More Information Needed]
 
1
  ---
2
  library_name: transformers
3
+ license: gpl-3.0
4
+ datasets:
5
+ - phunc20/nj_biergarten_captcha_v2
6
+ base_model:
7
+ - microsoft/trocr-base-handwritten
8
  ---
9
 
10
+ # Model Card for trocr-base-handwritten_nj_biergarten_captcha_v2
11
 
12
+ This is a model for CAPTCHA OCR.
13
 
14
 
15
 
 
17
 
18
  ### Model Description
19
 
20
+ This is a simple model finetuned from `microsoft/trocr-base-handwritten` on a dataset
21
+ I created at `phunc20/nj_biergarten_captcha_v2`.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
 
23
  ## Uses
24
 
25
  <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
26
 
27
+
28
  ### Direct Use
29
 
30
  <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
31
 
32
  [More Information Needed]
33
 
 
 
 
 
 
 
 
 
 
 
 
34
 
35
  ## Bias, Risks, and Limitations
36
 
37
+ Although the model seems to perform well on the dataset `phunc20/nj_biergarten_captcha_v2`,
38
+ it does not exhibit such good performance across all CAPTCHA images. In this respect, this
39
+ model is worse than Human.
40
 
41
  ### Recommendations
42
 
 
54
 
55
  ### Training Data
56
 
57
+ Like I mentioned, I trained this model on `phunc20/nj_biergarten_captcha_v2`.
58
+ In particular, I trained on the `train` split and evalaute on `validation` split,
59
+ without touching the `test` split.
60
 
61
  ### Training Procedure
62
 
63
+ Please refer to
64
+ <https://gitlab.com/phunc20/captchew/-/blob/main/colab_notebooks/train_from_pretrained_Seq2SeqTrainer_torchDataset.ipynb?ref_type=heads>
65
+ which is adapted from
66
+ <https://github.com/NielsRogge/Transformers-Tutorials/blob/master/TrOCR/Fine_tune_TrOCR_on_IAM_Handwriting_Database_using_Seq2SeqTrainer.ipynb>
 
 
 
 
 
 
 
 
 
 
 
 
67
 
68
  ## Evaluation
69
 
 
 
70
  ### Testing Data, Factors & Metrics
71
 
72
  #### Testing Data
73
 
74
+ 1. The `test` split of `phunc20/nj_biergarten_captcha_v2`
75
+ 2. This Kaggle dataset <https://www.kaggle.com/datasets/fournierp/captcha-version-2-images/data>
76
+ (we shall call this dataset by the name of `kaggle_test_set` in this model card.)
77
 
78
  #### Factors
79
 
 
83
 
84
  #### Metrics
85
 
86
+ CER, exact match and average length difference. The former two can be found in HuggingFace's
87
+ documentation. The last one is just one metric I care a little about. It is quite easy to
88
+ understand and, if need be, explanation could be found at the source code:
89
+ <https://gitlab.com/phunc20/captchew/-/blob/v0.1/average_length_difference.py>
90
 
91
  ### Results
92
+ On the `test` split of `phunc20/nj_biergarten_captcha_v2`
93
 
94
+ | Model | cer | exact match | avg len diff |
95
+ | --------------------------------------------------------- | -------- | ----------- | ------------ |
96
+ | `phunc20/trocr-base-handwritten_nj_biergarten_captcha_v2` | 0.001333 | 496/500 | 1/500 |
97
+ | `microsoft/trocr-base-handwritten` | 0.9 | 5/500 | 2.4 |
98
 
99
+ On `kaggle_test_set`
100
 
101
+ | Model | cer | exact match | avg len diff |
102
+ | --------------------------------------------------------- | -------- | ----------- | ------------ |
103
+ | `phunc20/trocr-base-handwritten_nj_biergarten_captcha_v2` | 0.4381 | 69/1070 | 0.1289 |
104
+ | `microsoft/trocr-base-handwritten` | 1.0112 | 17/1070 | 2.4439 |
105
 
 
 
 
 
 
106
 
107
  ## Environmental Impact
108
 
 
 
109
  Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
110
 
111
  - **Hardware Type:** [More Information Needed]
112
  - **Hours used:** [More Information Needed]
113
  - **Cloud Provider:** [More Information Needed]
114
  - **Compute Region:** [More Information Needed]
115
+ - **Carbon Emitted:** [More Information Needed]