def test_count_1_object_irregular():
a = np.array([[0., 0., 0., 0., 0.], [0., 1., 1., 1., 0.],
[0., 1., 1., 1., 1.], [0., 1., 1., 1., 0.],
[0., 0., 1., 0., 0.], [0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.]])
assert_equal(count_objects(a), 1)
def test_count_1_object_rectangle():
a = np.array([[0., 0., 0., 0., 0.], [0., 1., 1., 1., 0.],
[0., 1., 1., 1., 1.], [0., 1., 1., 1., 0.],
[0., 0., 1., 0., 0.], [0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.]])
assert_equal(count_objects(a), 1)
def test_count_objects_2_objects():
a = np.array([[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 1., 1., 1., 0., 0., 0., 0., 0., 0.],
[0., 1., 1., 1., 0., 0., 0., 0., 0., 0.],
[0., 1., 1., 1., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 1., 1., 0., 0., 0.],
[0., 0., 0., 0., 0., 1., 1., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]])
assert_equal(count_objects(a), 2)
from scipy.ndimage import imread
from counting import count_objects
if __name__ == '__main__':
img = imread('img/specimen_example.png', flatten=True) // 255
count = count_objects(img)
print('Number of objects:', count)
def test_count_no_object():
a = np.zeros([6, 6], dtype=np.float64)
assert_equal(count_objects(a), 0)
def test_count_all_ones():
a = np.ones([6, 6], dtype=np.int32)
assert_equal(count_objects(a), 1)
def test_count_1_object():
a = np.zeros([6, 6], dtype=np.float64)
a[1:-1, 1:-1] = 1
assert_equal(count_objects(a), 1)