Yiyao Wang
zihaomu
commited on
Commit
·
e6dbb4e
1
Parent(s):
84bf331
Fix the two FP16 models of CRNN(#131)
Browse files* The results of the two models remain the same accuracy.
* re-upload int8 models
* updated by yiyao
* remove the sim suffix
---------
Co-authored-by: zihaomu <[email protected]>
models/text_recognition_crnn/README.md
CHANGED
@@ -11,7 +11,7 @@ Results of accuracy evaluation with [tools/eval](../../tools/eval) at different
|
|
11 |
| CRNN_EN_INT8 | 81.75 | 75.33 | 52.43 |
|
12 |
| CRNN_CH | 71.28 | 80.90 | 67.36 |
|
13 |
| CRNN_CH_FP16 | 78.63 | 80.93 | 67.01 |
|
14 |
-
| CRNN_CH_INT8 | 78.
|
15 |
|
16 |
\*: 'FP16' or 'INT8' stands for 'model quantized into FP16' or 'model quantized into int8'
|
17 |
|
|
|
11 |
| CRNN_EN_INT8 | 81.75 | 75.33 | 52.43 |
|
12 |
| CRNN_CH | 71.28 | 80.90 | 67.36 |
|
13 |
| CRNN_CH_FP16 | 78.63 | 80.93 | 67.01 |
|
14 |
+
| CRNN_CH_INT8 | 78.11 | 81.20 | 67.01 |
|
15 |
|
16 |
\*: 'FP16' or 'INT8' stands for 'model quantized into FP16' or 'model quantized into int8'
|
17 |
|
tools/eval/datasets/icdar.py
CHANGED
@@ -41,7 +41,7 @@ class ICDAR:
|
|
41 |
|
42 |
rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])
|
43 |
pred = model.infer(img, rbbox)
|
44 |
-
if label == pred:
|
45 |
right_num += 1
|
46 |
|
47 |
self.acc = right_num/(len(self.val_label) * 1.0)
|
|
|
41 |
|
42 |
rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])
|
43 |
pred = model.infer(img, rbbox)
|
44 |
+
if label.lower() == pred.lower():
|
45 |
right_num += 1
|
46 |
|
47 |
self.acc = right_num/(len(self.val_label) * 1.0)
|