Mercurial > hg > nsaunier > traffic-intelligence

comparison python/ml.py @ 791:1158a6e2d28e dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
temporary solution for classification, with corrected svm.cpp and ml.hpp for loading saved classifiers
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 29 Apr 2016 16:07:35 -0400
parents 5b970a5bc233
children 180b6b0231c0
comparison
equal deleted inserted replaced
788:5b970a5bc233 791:1158a6e2d28e
18 ##################### 18 #####################
19 # OpenCV ML models 19 # OpenCV ML models
20 ##################### 20 #####################
21 21
22 class StatModel(object): 22 class StatModel(object):
23 '''Abstract class for loading/saving model''' 23 '''Abstract class for loading/saving model
24
25 Issues with OpenCV, does not seem to work'''
24 def load(self, filename): 26 def load(self, filename):
25 if path.exists(filename): 27 if path.exists(filename):
26 self.model.load(filename) 28 self.model.load(filename)
27 else: 29 else:
28 print('Provided filename {} does not exist: model not loaded!'.format(filename)) 30 print('Provided filename {} does not exist: model not loaded!'.format(filename))
40 self.model.setGamma(gamma) 42 self.model.setGamma(gamma)
41 self.model.setCoef0(coef0) 43 self.model.setCoef0(coef0)
42 self.model.setC(Cvalue) 44 self.model.setC(Cvalue)
43 self.model.setNu(nu) 45 self.model.setNu(nu)
44 self.model.setP(p) 46 self.model.setP(p)
47
48 def load(self, filename):
49 if path.exists(filename):
50 cv2.ml.SVM_load(filename)
51 else:
52 print('Provided filename {} does not exist: model not loaded!'.format(filename))
45 53
46 def train(self, samples, layout, responses): 54 def train(self, samples, layout, responses):
47 self.model.train(samples, layout, responses) 55 self.model.train(samples, layout, responses)
48 56
49 def predict(self, hog): 57 def predict(self, hog):