diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-14 22:15:47 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-09-14 22:15:47 +0200 |
| commit | 3b06ef615a8db67a03927576e0c12fbfb2501f5f (patch) | |
| tree | e1c2b1289971c8480327408de46152481e99b539 /src/text_recognizer/networks/ctc.py | |
| parent | 2b63fd952bdc9c7c72edd501cbcdbf3231e98f00 (diff) | |
Fixed CTC loss.
Diffstat (limited to 'src/text_recognizer/networks/ctc.py')
| -rw-r--r-- | src/text_recognizer/networks/ctc.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/text_recognizer/networks/ctc.py b/src/text_recognizer/networks/ctc.py index 72f18b8..2493d5c 100644 --- a/src/text_recognizer/networks/ctc.py +++ b/src/text_recognizer/networks/ctc.py @@ -24,7 +24,7 @@ def greedy_decoder( target_lengths (Optional[Tensor]): Length of each target tensor. Defaults to None. character_mapper (Optional[Callable]): A emnist/character mapper for mapping integers to characters. Defaults to None. - blank_label (int): The blank character to be ignored. Defaults to 79. + blank_label (int): The blank character to be ignored. Defaults to 80. collapse_repeated (bool): Collapase consecutive predictions of the same character. Defaults to True. Returns: |