samples: fix digits_video.py

maxgoedjen · Nov 8, 2018 · f30eafc · f30eafc
1 parent d9b8a9d
commit f30eafc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samples/python/digits_video.py b/samples/python/digits_video.py
@@ -86,7 +86,7 @@ def main():
                 frame[y:,x+w:][:SZ, :SZ] = bin_norm[...,np.newaxis]
 
             sample = preprocess_hog([bin_norm])
-            digit = model.predict(sample)[0]
+            digit = model.predict(sample)[1].ravel()
             cv.putText(frame, '%d'%digit, (x, y), cv.FONT_HERSHEY_PLAIN, 1.0, (200, 0, 0), thickness = 1)