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.03 | 80.93 | 66.67 |
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)