beam search: only normalize w.r.t. lengths if LM is used

githubharald · githubharald · commit 1892cc9d1f8e · 2018-06-20T19:00:25.000+02:00
diff --git a/src/BeamSearch.py b/src/BeamSearch.py
@@ -106,8 +106,9 @@ def ctcBeamSearch(mat, classes, lm):
 		# set new beam state
 		last = curr
 
-	# normalise probabilities according to labeling length
-	last.norm()
+	# normalise probabilities according to labeling length if LM is used
+	if lm:
+		last.norm()
 
 	 # sort by probability
 	bestLabeling = last.sort()[0] # get most probable labeling
