mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-03 08:02:00 +08:00
Vighnesh Birodkar
39 lines
891 B
Python
39 lines
891 B
Python
import numpy as np
|
|
from skimage.draw import circle
|
|
from skimage.feature import blob_dog
|
|
import math
|
|
|
|
|
|
def test_blob_dog():
|
|
img = np.ones((512, 512))
|
|
|
|
xs, ys = circle(400, 130, 5)
|
|
img[xs, ys] = 255
|
|
|
|
xs, ys = circle(100, 300, 25)
|
|
img[xs, ys] = 255
|
|
|
|
xs, ys = circle(200, 350, 45)
|
|
img[xs, ys] = 255
|
|
|
|
blobs = blob_dog(img, min_sigma=5, max_sigma=50)
|
|
area = lambda x: x[2]
|
|
radius = lambda x: math.sqrt(x / math.pi)
|
|
s = sorted(blobs, key=area)
|
|
thresh = 5
|
|
|
|
b = s[0]
|
|
assert abs(b[0] - 400) <= thresh
|
|
assert abs(b[1] - 130) <= thresh
|
|
assert abs(radius(b[2]) - 5) <= thresh
|
|
|
|
b = s[1]
|
|
assert abs(b[0] - 100) <= thresh
|
|
assert abs(b[1] - 300) <= thresh
|
|
assert abs(radius(b[2]) - 25) <= thresh
|
|
|
|
b = s[2]
|
|
assert abs(b[0] - 200) <= thresh
|
|
assert abs(b[1] - 350) <= thresh
|
|
assert abs(radius(b[2]) - 45) <= thresh
|