Merge branch 'master' of https://github.com/githubharald/CTCDecoder

githubharald · githubharald · commit 106ea6a6c890 · 2019-09-06T14:55:45.000+02:00
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+editdistance==0.5.2
+numpy==1.16.1
+pkg-resources==0.0.0
diff --git a/src/BeamSearch.py b/src/BeamSearch.py
@@ -49,12 +49,11 @@ def addBeam(beamState, labeling):
 		beamState.entries[labeling] = BeamEntry()
 
 
-def ctcBeamSearch(mat, classes, lm):
+def ctcBeamSearch(mat, classes, lm, beamWidth=25):
 	"beam search as described by the paper of Hwang et al. and the paper of Graves et al."
 
 	blankIdx = len(classes)
 	maxT, maxC = mat.shape
-	beamWidth = 25
 
 	# initialise beam state
 	last = BeamState()
