(image)
| 51 | |
| 52 | |
| 53 | def elec_detect(image): |
| 54 | hog_desc = get_hog_descriptor(test_img) |
| 55 | print(len(hog_desc)) |
| 56 | one_fv = np.zeros([len(hog_desc)], dtype=np.float32) |
| 57 | for i in range(len(hog_desc)): |
| 58 | one_fv[i] = hog_desc[i][0] |
| 59 | one_fv = np.reshape(one_fv, [-1, len(hog_desc)]) |
| 60 | print(len(one_fv), len(one_fv[0])) |
| 61 | svm = cv.ml.SVM_load('svm_data.dat') |
| 62 | result = svm.predict(one_fv)[1] |
| 63 | print(result) |
| 64 | |
| 65 | |
| 66 | if __name__ == '__main__': |
no test coverage detected