(positive_dir, negative_dir)
| 39 | |
| 40 | |
| 41 | def svm_train(positive_dir, negative_dir): |
| 42 | svm = cv.ml.SVM_create() |
| 43 | svm.setKernel(cv.ml.SVM_LINEAR) |
| 44 | svm.setType(cv.ml.SVM_C_SVC) |
| 45 | svm.setC(2.67) |
| 46 | svm.setGamma(5.383) |
| 47 | trainData, responses = get_dataset(positive_dir, negative_dir) |
| 48 | responses = np.reshape(responses, [-1, 1]) |
| 49 | svm.train(trainData, cv.ml.ROW_SAMPLE, responses) |
| 50 | svm.save('svm_data.dat') |
| 51 | |
| 52 | |
| 53 | def elec_detect(image): |
nothing calls this directly
no test coverage detected