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))