Initial implementation of Histograms of Oriented Gradients

This commit is contained in:
Brian Holt
2011-09-13 15:53:01 +01:00
parent bcd7069b7b
commit 2f897dc35d
3 changed files with 226 additions and 0 deletions
+25
View File
@@ -0,0 +1,25 @@
# Authors: Brian Holt
#
# License: BSD
import numpy as np
import scipy as sp
from scipy import ndimage
from numpy.testing import assert_raises
from scikits.image.feature import histogram_of_oriented_gradients
def test_histogram_of_oriented_gradients():
img = sp.lena().astype(np.int8)
fd, hog_image = histogram_of_oriented_gradients(img,
n_orientations=9,
ppc=(8,8),
cpb=(1,1),
visualise=False)
assert len(fd) == 9 * (512//8) ** 2
if __name__ == '__main__':
import nose
nose.runmodule()