digits_video.py (VERY IMMATURE)

digits.py now saves the resulting SVM classifier for use in digits_video.py
2012-06-27 05:59:16 +00:00
parent 13b30d7428
commit 6dffc5332d
2 changed files with 65 additions and 0 deletions
--- a/samples/python2/digits.py
+++ b/samples/python2/digits.py
@@ -124,5 +124,7 @@ if __name__ == '__main__':
    model.train(samples_train, labels_train)
    vis = evaluate_model(model, digits_test, samples_test, labels_test)
    cv2.imshow('SVM test', vis)
+    print 'saving SVM as "digits_svm.dat"...'
+    model.save('digits_svm.dat')

    cv2.waitKey(0)