From f30eafc9ac3ab7c31e468294d27429ae6e862f89 Mon Sep 17 00:00:00 2001 From: berak Date: Thu, 8 Nov 2018 09:17:41 +0100 Subject: [PATCH] samples: fix digits_video.py --- samples/python/digits_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samples/python/digits_video.py b/samples/python/digits_video.py index b2431c65fd..f976a32a28 100755 --- 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)