Merge pull request #7 from ThomasDelteil/patch-1

githubharald · web-flow · commit 0b952b42e499 · 2019-03-15T15:23:00.000+01:00
allow beamWidth as argument
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()
