forked from openvinotoolkit/open_model_zoo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
text-spotting-0001.yml
120 lines (96 loc) · 3.73 KB
/
text-spotting-0001.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
evaluations:
- name: text-spotting-0001
module: custom_evaluators.text_spotting_evaluator.TextSpottingEvaluator
module_config:
network_info:
detector:
model: intel/text-spotting-0001-detector/FP32/text-spotting-0001-detector.xml
weights: intel/text-spotting-0001-detector/FP32/text-spotting-0001-detector.bin
recognizer_encoder:
model: intel/text-spotting-0001-recognizer-encoder/FP32/text-spotting-0001-recognizer-encoder.xml
weights: intel/text-spotting-0001-recognizer-encoder/FP32/text-spotting-0001-recognizer-encoder.bin
recognizer_decoder:
model: intel/text-spotting-0001-recognizer-decoder/FP32/text-spotting-0001-recognizer-decoder.xml
weights: intel/text-spotting-0001-recognizer-decoder/FP32/text-spotting-0001-recognizer-decoder.bin
recognizer_decoder_inputs:
prev_symbol: prev_symbol
prev_hidden: prev_hidden
encoder_outputs: encoder_outputs
recognizer_decoder_outputs:
symbols_distribution: output
cur_hidden: hidden
max_seq_len: '28'
alphabet: __0123456789abcdefghijklmnopqrstuvwxyz
sos_index: '0'
eos_index: '1'
adapter:
type: mask_rcnn_with_text
classes_out: classes
scores_out: scores
boxes_out: boxes
raw_masks_out: raw_masks
texts_out: texts
confidence_threshold: 0.8
launchers:
- framework: dlsdk
tags:
- FP32
datasets:
- name: ICDAR2015_word_spotting
preprocessing:
- type: resize
dst_width: 1280
dst_height: 768
metrics:
- type: incidental_text_hmean
name: f-measure
ignore_difficult: True
word_spotting: True
- name: text-spotting-0001
module: custom_evaluators.text_spotting_evaluator.TextSpottingEvaluator
module_config:
network_info:
detector:
model: intel/text-spotting-0001-detector/FP16/text-spotting-0001-detector.xml
weights: intel/text-spotting-0001-detector/FP16/text-spotting-0001-detector.bin
recognizer_encoder:
model: intel/text-spotting-0001-recognizer-encoder/FP16/text-spotting-0001-recognizer-encoder.xml
weights: intel/text-spotting-0001-recognizer-encoder/FP16/text-spotting-0001-recognizer-encoder.bin
recognizer_decoder:
model: intel/text-spotting-0001-recognizer-decoder/FP16/text-spotting-0001-recognizer-decoder.xml
weights: intel/text-spotting-0001-recognizer-decoder/FP16/text-spotting-0001-recognizer-decoder.bin
recognizer_decoder_inputs:
prev_symbol: prev_symbol
prev_hidden: prev_hidden
encoder_outputs: encoder_outputs
recognizer_decoder_outputs:
symbols_distribution: output
cur_hidden: hidden
max_seq_len: '28'
alphabet: __0123456789abcdefghijklmnopqrstuvwxyz
sos_index: '0'
eos_index: '1'
adapter:
type: mask_rcnn_with_text
classes_out: classes
scores_out: scores
boxes_out: boxes
raw_masks_out: raw_masks
texts_out: texts
confidence_threshold: 0.8
launchers:
- framework: dlsdk
tags:
- FP16
datasets:
- name: ICDAR2015_word_spotting
preprocessing:
- type: resize
dst_width: 1280
dst_height: 768
metrics:
- type: incidental_text_hmean
name: f-measure
ignore_difficult: True
word_spotting: True
global_definitions: ../dataset_definitions.yml