generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix digits.py to not ignore model parameters

Browse Source 
otherwise model is only ever trained with default
parameters, and digits_adjust does nothing (for example)
This commit is contained in:
John Stowers 2015-07-31 09:14:20 +02:00
parent 429d280501
commit c4c3f1b25e
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
samples/python2/digits.py
View File

@ -77,7 +77,6 @@ class KNearest(StatModel):
self.model = cv2.ml.KNearest_create() self.model = cv2.ml.KNearest_create()
def train(self, samples, responses): def train(self, samples, responses):
self.model = cv2.ml.KNearest_create()
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses) self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples): def predict(self, samples):
@ -93,7 +92,6 @@ class SVM(StatModel):
self.model.setType(cv2.ml.SVM_C_SVC) self.model.setType(cv2.ml.SVM_C_SVC)
def train(self, samples, responses): def train(self, samples, responses):
self.model = cv2.ml.SVM_create()
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses) self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples): def predict(self, samples):