Improvements in tests, bugfix in digits_video sample

2016-02-05 17:46:52 +03:00
parent 4e3a6328ba
commit 87fc75c6d7
8 changed files with 62 additions and 50 deletions
--- a/modules/python/test/test_digits.py
+++ b/modules/python/test/test_digits.py
@@ -95,7 +95,7 @@ def evaluate_model(model, digits, samples, labels):

    confusion = np.zeros((10, 10), np.int32)
    for i, j in zip(labels, resp):
-        confusion[i, j] += 1
+        confusion[int(i), int(j)] += 1

    return err, confusion