Mercurial Hosting > traffic-intelligence
comparison python/ml.py @ 791:1158a6e2d28e dev
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): |