mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-28 13:10:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #10663 from jmlich:master

Browse Source 
* hogsvm compatibility with python3
This commit is contained in:
Jozef Mlich 2018-01-24 15:03:23 +01:00 committed by Alexander Alekhin
parent b97b650ab3
commit 7a472d85ef
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
samples/python/tutorial_code/ml/py_svm_opencv/hogsvm.py Normal file → Executable file
View File

@ -1,3 +1,5 @@
#!/usr/bin/env python
import cv2 as cv
import numpy as np
@ -44,8 +46,8 @@ test_cells = [ i[50:] for i in cells]
###### Now training ########################
deskewed = [map(deskew,row) for row in train_cells]
hogdata = [map(hog,row) for row in deskewed]
deskewed = [list(map(deskew,row)) for row in train_cells]
hogdata = [list(map(hog,row)) for row in deskewed]
trainData = np.float32(hogdata).reshape(-1,64)
responses = np.repeat(np.arange(10),250)[:,np.newaxis]
@ -60,12 +62,12 @@ svm.save('svm_data.dat')
###### Now testing ########################
deskewed = [map(deskew,row) for row in test_cells]
hogdata = [map(hog,row) for row in deskewed]
deskewed = [list(map(deskew,row)) for row in test_cells]
hogdata = [list(map(hog,row)) for row in deskewed]
testData = np.float32(hogdata).reshape(-1,bin_n*4)
result = svm.predict(testData)[1]
####### Check Accuracy ########################
mask = result==responses
correct = np.count_nonzero(mask)
print correct*100.0/result.size
print(correct*100.0/result.size)