githubharald / CTCDecoder

Connectionist Temporal Classification (CTC) decoding algorithms: best path, beam search, lexicon search, prefix search, and token passing. Implemented in Python.
https://towardsdatascience.com/3797e43a86c
MIT License
817 stars 182 forks source link

beam_search.py don't support batch data #8

Closed huangnengCSU closed 5 years ago

huangnengCSU commented 5 years ago

def ctcBeamSearch(mat, classes, lm, beamWidth=25): blankIdx = len(classes) maxT, maxC = mat.shape the matrix only have 2 dimension(length, char_size), don't have batch dimension.

githubharald commented 5 years ago

This observation is correct. Nothing to do here - therefore closing this issue.