fix crnn postprocess error

This commit is contained in:
chenhaozhe 2021-07-22 14:48:07 +08:00
parent e1e7d491d9
commit 4baa370165
2 changed files with 9 additions and 4 deletions

View File

@ -39,6 +39,11 @@ CRNN was a neural network for image based sequence recognition and its Applicati
CRNN use a vgg16 structure for feature extraction, the appending with two-layer bidirectional LSTM, finally use CTC to calculate loss. See src/crnn.py for details.
We provide 2 versions of network using different ways to transfer the hidden size to class numbers. You could choose different version by modifying the `model_version` in config yaml.
- V1 using an full connection after the RNN parts.
- V2 change the output feature size of the last RNN, to output a feature with the same size of class numbers.
## [Dataset](#content)
Note that you can run the scripts based on the dataset mentioned in original paper or widely used in relevant domain/network architecture. In the following sections, we will introduce how to run the scripts using the related dataset below.

View File

@ -8,14 +8,14 @@ checkpoint_url: ""
data_path: "/cache/data"
output_path: "/cache/train"
load_path: "/cache/checkpoint_path"
device_target: "GPU"
device_target: "Ascend"
enable_profiling: False
# ======================================================================================
# common options
run_distribute: False
model: "lowercase"
model_version: "V2"
model_version: "V1"
# ======================================================================================
# Training options
@ -64,8 +64,8 @@ file_format: "MINDIR"
# ======================================================================================
#postprocess
ann_file: True
result_path: True
ann_file: ""
result_path: ""
dataset: "ic03"
---