def test_local_max(self):
bd = BlobDetection(self.img)
bd._one_octave(shrink=False, refine=False, n_5=False)
self.assert_(numpy.alltrue(_blob.local_max(bd.dogs, bd.cur_mask, False) == \
local_max(bd.dogs, bd.cur_mask, False)), "max test, 3x3x3")
self.assert_(numpy.alltrue(_blob.local_max(bd.dogs, bd.cur_mask, True) == \
local_max(bd.dogs, bd.cur_mask, True)), "max test, 3x5x5")
def test_local_max(self):
bd = BlobDetection(self.img)
bd._one_octave(shrink=False, refine=False, n_5=False)
self.assert_(numpy.alltrue(_blob.local_max(bd.dogs, bd.cur_mask, False) == \
local_max(bd.dogs, bd.cur_mask, False)), "max test, 3x3x3")
self.assert_(numpy.alltrue(_blob.local_max(bd.dogs, bd.cur_mask, True) == \
local_max(bd.dogs, bd.cur_mask, True)), "max test, 3x5x5")
msk = None
else:
data = image_test_rings()
msk = None
bd = BlobDetection(data, mask=msk) # , cur_sigma=0.25, init_sigma=numpy.sqrt(2)/2, dest_sigma=numpy.sqrt(2), scale_per_octave=2)
pylab.ion()
f = pylab.figure(1)
ax = f.add_subplot(111)
ax.imshow(numpy.log1p(data), interpolation='nearest')
for i in range(5):
print('Octave #%i' % i)
bd._one_octave(shrink=True, refine=True, n_5=False)
print("Octave #%i Total kp: %i" % (i, bd.keypoints.size))
# bd._one_octave(False, True ,False)
print('Final size of keypoints : %i' % bd.keypoints.size)
i = 0
# for kp in bd.keypoints:
# ds = kp.sigma
# ax.annotate("", xy=(kp.x, kp.y), xytext=(kp.x+ds, kp.y+ds),
# arrowprops=dict(facecolor='blue', shrink=0.05),)
sigma = bd.keypoints.sigma
for i, c in enumerate("ygrcmykw"):
# j = 2 ** i
m = numpy.logical_and(sigma >= i, sigma < (i + 1))
else:
data = image_test_rings()
msk = None
bd = BlobDetection(
data, mask=msk
) # , cur_sigma=0.25, init_sigma=numpy.sqrt(2)/2, dest_sigma=numpy.sqrt(2), scale_per_octave=2)
pylab.ion()
f = pylab.figure(1)
ax = f.add_subplot(111)
ax.imshow(numpy.log1p(data), interpolation='nearest')
for i in range(5):
print('Octave #%i' % i)
bd._one_octave(shrink=True, refine=True, n_5=False)
print("Octave #%i Total kp: %i" % (i, bd.keypoints.size))
# bd._one_octave(False, True ,False)
print('Final size of keypoints : %i' % bd.keypoints.size)
i = 0
# for kp in bd.keypoints:
# ds = kp.sigma
# ax.annotate("", xy=(kp.x, kp.y), xytext=(kp.x+ds, kp.y+ds),
# arrowprops=dict(facecolor='blue', shrink=0.05),)
sigma = bd.keypoints.sigma
for i, c in enumerate("ygrcmykw"):
# j = 2 ** i
m = numpy.logical_and(sigma >= i, sigma < (i + 1))
