Merge pull request #13076 from berak:fix_digits_video_py

7 years ago · 4b298a7460
parent 6adbbcdde1 f30eafc9ac
commit 4b298a7460
1 changed files with 1 additions and 1 deletions
--- 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)