1112 lines
38 KiB
Python
1112 lines
38 KiB
Python
|
import numpy as np
|
||
|
|
from skimage._shared._warnings import expected_warnings
|
||
|
|
from skimage._shared.testing import test_parallel
|
||
|
|
from skimage._shared import testing
|
||
|
|
from skimage._shared.testing import assert_array_equal, assert_equal
|
||
|
|
from skimage._shared.testing import assert_almost_equal
|
||
|
|
|
||
|
|
from skimage.draw import (set_color, line, line_aa, polygon, polygon_perimeter,
|
||
|
|
disk,
|
||
|
|
circle, circle_perimeter, circle_perimeter_aa,
|
||
|
|
ellipse, ellipse_perimeter,
|
||
|
|
_bezier_segment, bezier_curve, rectangle,
|
||
|
|
rectangle_perimeter)
|
||
|
|
from skimage.measure import regionprops
|
||
|
|
|
||
|
|
|
||
|
|
def test_set_color():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc = line(0, 0, 0, 30)
|
||
|
|
set_color(img, (rr, cc), 1)
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[0, :] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_set_color_with_alpha():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc, alpha = line_aa(0, 0, 0, 30)
|
||
|
|
set_color(img, (rr, cc), 1, alpha=alpha)
|
||
|
|
|
||
|
|
# Wrong dimensionality color
|
||
|
|
with testing.raises(ValueError):
|
||
|
|
set_color(img, (rr, cc), (255, 0, 0), alpha=alpha)
|
||
|
|
|
||
|
|
img = np.zeros((10, 10, 3))
|
||
|
|
|
||
|
|
rr, cc, alpha = line_aa(0, 0, 0, 30)
|
||
|
|
set_color(img, (rr, cc), (1, 0, 0), alpha=alpha)
|
||
|
|
|
||
|
|
|
||
|
|
@test_parallel()
|
||
|
|
def test_line_horizontal():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc = line(0, 0, 0, 9)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[0, :] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_vertical():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc = line(0, 0, 9, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[:, 0] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_reverse():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc = line(0, 9, 0, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[0, :] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_diag():
|
||
|
|
img = np.zeros((5, 5))
|
||
|
|
|
||
|
|
rr, cc = line(0, 0, 4, 4)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.eye(5)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_aa_horizontal():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc, val = line_aa(0, 0, 0, 9)
|
||
|
|
set_color(img, (rr, cc), 1, alpha=val)
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[0, :] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_aa_vertical():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc, val = line_aa(0, 0, 9, 0)
|
||
|
|
img[rr, cc] = val
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[:, 0] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_aa_diagonal():
|
||
|
|
img = np.zeros((10, 10))
|
||
|
|
|
||
|
|
rr, cc, val = line_aa(0, 0, 9, 6)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
# Check that each pixel belonging to line,
|
||
|
|
# also belongs to line_aa
|
||
|
|
r, c = line(0, 0, 9, 6)
|
||
|
|
for r_i, c_i in zip(r, c):
|
||
|
|
assert_equal(img[r_i, c_i], 1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_line_equal_aliasing_horizontally_vertically():
|
||
|
|
img0 = np.zeros((25, 25))
|
||
|
|
img1 = np.zeros((25, 25))
|
||
|
|
|
||
|
|
# Near-horizontal line
|
||
|
|
rr, cc, val = line_aa(10, 2, 12, 20)
|
||
|
|
img0[rr, cc] = val
|
||
|
|
|
||
|
|
# Near-vertical (transpose of prior)
|
||
|
|
rr, cc, val = line_aa(2, 10, 20, 12)
|
||
|
|
img1[rr, cc] = val
|
||
|
|
|
||
|
|
# Difference - should be zero
|
||
|
|
assert_array_equal(img0, img1.T)
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_rectangle():
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
|
||
|
|
rr, cc = polygon((1, 4, 4, 1, 1), (1, 1, 4, 4, 1))
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[1:4, 1:4] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_rectangle_angular():
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
poly = np.array(((0, 3), (4, 7), (7, 4), (3, 0), (0, 3)))
|
||
|
|
|
||
|
|
rr, cc = polygon(poly[:, 0], poly[:, 1])
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1, 0, 0, 0, 0, 0],
|
||
|
|
[1, 1, 1, 1, 1, 1, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 1, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 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]]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_parallelogram():
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
poly = np.array(((1, 1), (5, 1), (7, 6), (3, 6), (1, 1)))
|
||
|
|
|
||
|
|
rr, cc = polygon(poly[:, 0], poly[:, 1])
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 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]]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_exceed():
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
poly = np.array(((1, -1), (100, -1), (100, 100), (1, 100), (1, 1)))
|
||
|
|
|
||
|
|
rr, cc = polygon(poly[:, 0], poly[:, 1], img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros((10, 10))
|
||
|
|
img_[1:, :] = 1
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_deprecated():
|
||
|
|
with expected_warnings(['circle is deprecated']):
|
||
|
|
_ = circle(7, 7, 6)
|
||
|
|
|
||
|
|
|
||
|
|
def test_disk():
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
|
||
|
|
rr, cc = disk((7, 7), 6)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.array(
|
||
|
|
[[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, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 1, 1, 1, 1, 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]]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_perimeter_bresenham():
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7, 7, 0, method='bresenham')
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert(np.sum(img) == 1)
|
||
|
|
assert(img[7][7] == 1)
|
||
|
|
|
||
|
|
img = np.zeros((17, 15), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7, 7, 7, method='bresenham')
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 1, 1, 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]]
|
||
|
|
)
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_perimeter_bresenham_shape():
|
||
|
|
img = np.zeros((15, 20), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7, 10, 9, method='bresenham', shape=(15, 20))
|
||
|
|
img[rr, cc] = 1
|
||
|
|
shift = 5
|
||
|
|
img_ = np.zeros((15 + 2 * shift, 20), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7 + shift, 10, 9, method='bresenham', shape=None)
|
||
|
|
img_[rr, cc] = 1
|
||
|
|
assert_array_equal(img, img_[shift:-shift, :])
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_perimeter_andres():
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7, 7, 0, method='andres')
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert(np.sum(img) == 1)
|
||
|
|
assert(img[7][7] == 1)
|
||
|
|
|
||
|
|
img = np.zeros((17, 15), 'uint8')
|
||
|
|
rr, cc = circle_perimeter(7, 7, 7, method='andres')
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0],
|
||
|
|
[0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
|
||
|
|
[1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 1, 1, 1, 1, 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]]
|
||
|
|
)
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_perimeter_aa():
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
rr, cc, val = circle_perimeter_aa(7, 7, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert(np.sum(img) == 1)
|
||
|
|
assert(img[7][7] == 1)
|
||
|
|
|
||
|
|
img = np.zeros((17, 17), 'uint8')
|
||
|
|
rr, cc, val = circle_perimeter_aa(8, 8, 7)
|
||
|
|
img[rr, cc] = val * 255
|
||
|
|
img_ = np.array(
|
||
|
|
[[ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 0, 0, 82, 180, 236, 255, 236, 180, 82, 0, 0, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 0, 189, 172, 74, 18, 0, 18, 74, 172, 189, 0, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 229, 25, 0, 0, 0, 0, 0, 0, 0, 25, 229, 0, 0, 0],
|
||
|
|
[ 0, 0, 189, 25, 0, 0, 0, 0, 0, 0, 0, 0, 0, 25, 189, 0, 0],
|
||
|
|
[ 0, 82, 172, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 172, 82, 0],
|
||
|
|
[ 0, 180, 74, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 74, 180, 0],
|
||
|
|
[ 0, 236, 18, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 18, 236, 0],
|
||
|
|
[ 0, 255, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 0],
|
||
|
|
[ 0, 236, 18, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 18, 236, 0],
|
||
|
|
[ 0, 180, 74, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 74, 180, 0],
|
||
|
|
[ 0, 82, 172, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 172, 82, 0],
|
||
|
|
[ 0, 0, 189, 25, 0, 0, 0, 0, 0, 0, 0, 0, 0, 25, 189, 0, 0],
|
||
|
|
[ 0, 0, 0, 229, 25, 0, 0, 0, 0, 0, 0, 0, 25, 229, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 0, 189, 172, 74, 18, 0, 18, 74, 172, 189, 0, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 0, 0, 82, 180, 236, 255, 236, 180, 82, 0, 0, 0, 0, 0],
|
||
|
|
[ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
|
||
|
|
)
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_circle_perimeter_aa_shape():
|
||
|
|
img = np.zeros((15, 20), 'uint8')
|
||
|
|
rr, cc, val = circle_perimeter_aa(7, 10, 9, shape=(15, 20))
|
||
|
|
img[rr, cc] = val * 255
|
||
|
|
|
||
|
|
shift = 5
|
||
|
|
img_ = np.zeros((15 + 2 * shift, 20), 'uint8')
|
||
|
|
rr, cc, val = circle_perimeter_aa(7 + shift, 10, 9, shape=None)
|
||
|
|
img_[rr, cc] = val * 255
|
||
|
|
assert_array_equal(img, img_[shift:-shift, :])
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_trivial():
|
||
|
|
img = np.zeros((2, 2), 'uint8')
|
||
|
|
rr, cc = ellipse(0.5, 0.5, 0.5, 0.5)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_correct = np.array([
|
||
|
|
[0, 0],
|
||
|
|
[0, 0]
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_correct)
|
||
|
|
|
||
|
|
img = np.zeros((2, 2), 'uint8')
|
||
|
|
rr, cc = ellipse(0.5, 0.5, 1.1, 1.1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_correct = np.array([
|
||
|
|
[1, 1],
|
||
|
|
[1, 1],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_correct)
|
||
|
|
|
||
|
|
img = np.zeros((3, 3), 'uint8')
|
||
|
|
rr, cc = ellipse(1, 1, 0.9, 0.9)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_correct = np.array([
|
||
|
|
[0, 0, 0],
|
||
|
|
[0, 1, 0],
|
||
|
|
[0, 0, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_correct)
|
||
|
|
|
||
|
|
img = np.zeros((3, 3), 'uint8')
|
||
|
|
rr, cc = ellipse(1, 1, 1.1, 1.1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_correct = np.array([
|
||
|
|
[0, 1, 0],
|
||
|
|
[1, 1, 1],
|
||
|
|
[0, 1, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_correct)
|
||
|
|
|
||
|
|
img = np.zeros((3, 3), 'uint8')
|
||
|
|
rr, cc = ellipse(1, 1, 1.5, 1.5)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_correct = np.array([
|
||
|
|
[1, 1, 1],
|
||
|
|
[1, 1, 1],
|
||
|
|
[1, 1, 1],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_correct)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_generic():
|
||
|
|
img = np.zeros((4, 4), 'uint8')
|
||
|
|
rr, cc = ellipse(1.5, 1.5, 1.1, 1.7)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array([
|
||
|
|
[0, 0, 0, 0],
|
||
|
|
[1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1],
|
||
|
|
[0, 0, 0, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
img = np.zeros((5, 5), 'uint8')
|
||
|
|
rr, cc = ellipse(2, 2, 1.7, 1.7)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array([
|
||
|
|
[0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
rr, cc = ellipse(5, 5, 3, 4)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array([
|
||
|
|
[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, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
img = np.zeros((10, 10), 'uint8')
|
||
|
|
rr, cc = ellipse(4.5, 5, 3.5, 4)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array([
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
])
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
rr, cc = ellipse(7, 7, 3, 7)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array([
|
||
|
|
[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, 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, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 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, 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_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_with_shape():
|
||
|
|
img = np.zeros((15, 15), 'uint8')
|
||
|
|
|
||
|
|
rr, cc = ellipse(7, 7, 3, 10, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
img_ = np.array(
|
||
|
|
[[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, 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],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 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, 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_array_equal(img, img_)
|
||
|
|
|
||
|
|
img = np.zeros((10, 9, 3), 'uint8')
|
||
|
|
|
||
|
|
rr, cc = ellipse(7, 7, 3, 10, shape=img.shape)
|
||
|
|
img[rr, cc, 0] = 1
|
||
|
|
|
||
|
|
img_ = np.zeros_like(img)
|
||
|
|
img_[..., 0] = np.array(
|
||
|
|
[[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, 0, 0],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1],
|
||
|
|
[1, 1, 1, 1, 1, 1, 1, 1, 1]],
|
||
|
|
)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_negative():
|
||
|
|
rr, cc = ellipse(-3, -3, 1.7, 1.7)
|
||
|
|
rr_, cc_ = np.nonzero(np.array([
|
||
|
|
[0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0],
|
||
|
|
]))
|
||
|
|
|
||
|
|
assert_array_equal(rr, rr_ - 5)
|
||
|
|
assert_array_equal(cc, cc_ - 5)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_rotation_symmetry():
|
||
|
|
img1 = np.zeros((150, 150), dtype=np.uint8)
|
||
|
|
img2 = np.zeros((150, 150), dtype=np.uint8)
|
||
|
|
for angle in range(0, 180, 15):
|
||
|
|
img1.fill(0)
|
||
|
|
rr, cc = ellipse(80, 70, 60, 40, rotation=np.deg2rad(angle))
|
||
|
|
img1[rr, cc] = 1
|
||
|
|
img2.fill(0)
|
||
|
|
rr, cc = ellipse(80, 70, 60, 40, rotation=np.deg2rad(angle + 180))
|
||
|
|
img2[rr, cc] = 1
|
||
|
|
assert_array_equal(img1, img2)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_rotated():
|
||
|
|
img = np.zeros((1000, 1200), dtype=np.uint8)
|
||
|
|
for rot in range(0, 180, 10):
|
||
|
|
img.fill(0)
|
||
|
|
angle = np.deg2rad(rot)
|
||
|
|
rr, cc = ellipse(500, 600, 200, 400, rotation=angle)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
# estimate orientation of ellipse
|
||
|
|
angle_estim_raw = regionprops(img)[0].orientation
|
||
|
|
angle_estim = np.round(angle_estim_raw, 3) % (np.pi / 2)
|
||
|
|
assert_almost_equal(angle_estim, angle % (np.pi / 2), 2)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_dot_zeroangle():
|
||
|
|
# dot, angle == 0
|
||
|
|
img = np.zeros((30, 15), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(15, 7, 0, 0, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert(np.sum(img) == 1)
|
||
|
|
assert(img[15][7] == 1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_dot_nzeroangle():
|
||
|
|
# dot, angle != 0
|
||
|
|
img = np.zeros((30, 15), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(15, 7, 0, 0, 1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert(np.sum(img) == 1)
|
||
|
|
assert(img[15][7] == 1)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_flat_zeroangle():
|
||
|
|
# flat ellipse
|
||
|
|
img = np.zeros((20, 18), 'uint8')
|
||
|
|
img_ = np.zeros((20, 18), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(6, 7, 0, 5, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
rr, cc = line(6, 2, 6, 12)
|
||
|
|
img_[rr, cc] = 1
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_zeroangle():
|
||
|
|
# angle == 0
|
||
|
|
img = np.zeros((30, 15), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(15, 7, 14, 6, 0)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0]]
|
||
|
|
)
|
||
|
|
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_nzeroangle():
|
||
|
|
# angle != 0
|
||
|
|
img = np.zeros((30, 25), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(15, 11, 12, 6, 1.1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array(
|
||
|
|
[[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, 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, 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, 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, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 1, 1, 1, 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, 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, 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, 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_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ellipse_perimeter_shape():
|
||
|
|
img = np.zeros((15, 20), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(7, 10, 9, 9, 0, shape=(15, 20))
|
||
|
|
img[rr, cc] = 1
|
||
|
|
shift = 5
|
||
|
|
img_ = np.zeros((15 + 2 * shift, 20), 'uint8')
|
||
|
|
rr, cc = ellipse_perimeter(7 + shift, 10, 9, 9, 0, shape=None)
|
||
|
|
img_[rr, cc] = 1
|
||
|
|
assert_array_equal(img, img_[shift:-shift, :])
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_segment_straight():
|
||
|
|
image = np.zeros((200, 200), dtype=int)
|
||
|
|
r0, r1, r2 = 50, 150, 150
|
||
|
|
c0, c1, c2 = 50, 50, 150
|
||
|
|
rr, cc = _bezier_segment(r0, c0, r1, c1, r2, c2, 0)
|
||
|
|
image[rr, cc] = 1
|
||
|
|
|
||
|
|
image2 = np.zeros((200, 200), dtype=int)
|
||
|
|
rr, cc = line(r0, c0, r2, c2)
|
||
|
|
image2[rr, cc] = 1
|
||
|
|
assert_array_equal(image, image2)
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_segment_curved():
|
||
|
|
img = np.zeros((25, 25), 'uint8')
|
||
|
|
r0, c0 = 20, 20
|
||
|
|
r1, c1 = 20, 2
|
||
|
|
r2, c2 = 2, 2
|
||
|
|
rr, cc = _bezier_segment(r0, c0, r1, c1, r2, c2, 1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
img_ = np.array(
|
||
|
|
[[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, 0, 0, 0, 0, 0, 0, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 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, 1, 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, 1, 1, 1, 1, 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, 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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
|
||
|
|
)
|
||
|
|
assert_equal(img[r0, c0], 1)
|
||
|
|
assert_equal(img[r2, c2], 1)
|
||
|
|
assert_array_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_curve_straight():
|
||
|
|
image = np.zeros((200, 200), dtype=int)
|
||
|
|
r0, c0 = 50, 50
|
||
|
|
r1, c1 = 150, 50
|
||
|
|
r2, c2 = 150, 150
|
||
|
|
rr, cc = bezier_curve(r0, c0, r1, c1, r2, c2, 0)
|
||
|
|
image[rr, cc] = 1
|
||
|
|
|
||
|
|
image2 = np.zeros((200, 200), dtype=int)
|
||
|
|
rr, cc = line(r0, c0, r2, c2)
|
||
|
|
image2[rr, cc] = 1
|
||
|
|
assert_array_equal(image, image2)
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_curved_weight_eq_1():
|
||
|
|
img = np.zeros((23, 8), 'uint8')
|
||
|
|
r0, c0 = 1, 1
|
||
|
|
r1, c1 = 11, 11
|
||
|
|
r2, c2 = 21, 1
|
||
|
|
rr, cc = bezier_curve(r0, c0, r1, c1, r2, c2, 1)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_equal(img[r0, c0], 1)
|
||
|
|
assert_equal(img[r2, c2], 1)
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0]]
|
||
|
|
)
|
||
|
|
assert_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_curved_weight_neq_1():
|
||
|
|
img = np.zeros((23, 10), 'uint8')
|
||
|
|
r0, c0 = 1, 1
|
||
|
|
r1, c1 = 11, 11
|
||
|
|
r2, c2 = 21, 1
|
||
|
|
rr, cc = bezier_curve(r0, c0, r1, c1, r2, c2, 2)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_equal(img[r0, c0], 1)
|
||
|
|
assert_equal(img[r2, c2], 1)
|
||
|
|
img_ = np.array(
|
||
|
|
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 1, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 1, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 1, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 1, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
|
||
|
|
)
|
||
|
|
assert_equal(img, img_)
|
||
|
|
|
||
|
|
|
||
|
|
def test_bezier_curve_shape():
|
||
|
|
img = np.zeros((15, 20), 'uint8')
|
||
|
|
r0, c0 = 1, 5
|
||
|
|
r1, c1 = 6, 11
|
||
|
|
r2, c2 = 1, 14
|
||
|
|
rr, cc = bezier_curve(r0, c0, r1, c1, r2, c2, 2, shape=(15, 20))
|
||
|
|
img[rr, cc] = 1
|
||
|
|
shift = 5
|
||
|
|
img_ = np.zeros((15 + 2 * shift, 20), 'uint8')
|
||
|
|
r0, c0 = 1 + shift, 5
|
||
|
|
r1, c1 = 6 + shift, 11
|
||
|
|
r2, c2 = 1 + shift, 14
|
||
|
|
rr, cc = bezier_curve(r0, c0, r1, c1, r2, c2, 2, shape=None)
|
||
|
|
img_[rr, cc] = 1
|
||
|
|
assert_array_equal(img, img_[shift:-shift, :])
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_perimeter():
|
||
|
|
expected = np.array(
|
||
|
|
[[1, 1, 1, 1],
|
||
|
|
[1, 0, 0, 1],
|
||
|
|
[1, 1, 1, 1]]
|
||
|
|
)
|
||
|
|
out = np.zeros_like(expected)
|
||
|
|
|
||
|
|
rr, cc = polygon_perimeter([0, 2, 2, 0],
|
||
|
|
[0, 0, 3, 3])
|
||
|
|
|
||
|
|
out[rr, cc] = 1
|
||
|
|
assert_array_equal(out, expected)
|
||
|
|
|
||
|
|
out = np.zeros_like(expected)
|
||
|
|
rr, cc = polygon_perimeter([-1, -1, 3, 3],
|
||
|
|
[-1, 4, 4, -1],
|
||
|
|
shape=out.shape, clip=True)
|
||
|
|
out[rr, cc] = 1
|
||
|
|
assert_array_equal(out, expected)
|
||
|
|
|
||
|
|
with testing.raises(ValueError):
|
||
|
|
polygon_perimeter([0], [1], clip=True)
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon_perimeter_outside_image():
|
||
|
|
rr, cc = polygon_perimeter([-1, -1, 3, 3],
|
||
|
|
[-1, 4, 4, -1], shape=(3, 4))
|
||
|
|
assert_equal(len(rr), 0)
|
||
|
|
assert_equal(len(cc), 0)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_end():
|
||
|
|
expected = np.array([[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
start = (0, 1)
|
||
|
|
end = (3, 3)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(start, end=end, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# Swap start and end
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(end=start, start=end, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# Bottom left and top right
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(start=(3, 1), end=(0, 3), shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(end=(3, 1), start=(0, 3), shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_extent():
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
start = (1, 1)
|
||
|
|
extent = (3, 3)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
img = np.zeros((5, 5, 3), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc, 0] = 1
|
||
|
|
expected_2 = np.zeros_like(img)
|
||
|
|
expected_2[..., 0] = expected
|
||
|
|
assert_array_equal(img, expected_2)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_extent_negative():
|
||
|
|
# These two tests should be done together.
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 1, 2, 2, 1],
|
||
|
|
[0, 0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
|
||
|
|
start = (3, 5)
|
||
|
|
extent = (-1, -2)
|
||
|
|
img = np.zeros(expected.shape, dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
|
||
|
|
rr, cc = rectangle(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 2
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# Ensure that rr and cc have no overlap
|
||
|
|
img = np.zeros(expected.shape, dtype=np.uint8)
|
||
|
|
rr, cc = rectangle(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 2
|
||
|
|
|
||
|
|
rr, cc = rectangle_perimeter(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_perimiter():
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0, 0],
|
||
|
|
[0, 0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 1, 0, 0, 1],
|
||
|
|
[0, 0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
start = (2, 3)
|
||
|
|
end = (2, 4)
|
||
|
|
img = np.zeros(expected.shape, dtype=np.uint8)
|
||
|
|
# Test that the default parameter is indeed end
|
||
|
|
rr, cc = rectangle_perimeter(start, end, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# Swap start and end
|
||
|
|
img = np.zeros(expected.shape, dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(end=start, start=end, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
img = np.zeros(expected.shape, dtype=np.uint8)
|
||
|
|
start = (2, 3)
|
||
|
|
extent = (1, 2)
|
||
|
|
rr, cc = rectangle_perimeter(start, extent=extent, shape=img.shape)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_perimiter_clip_bottom_right():
|
||
|
|
# clip=False
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1],
|
||
|
|
[0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
start = (2, 2)
|
||
|
|
extent = (10, 10)
|
||
|
|
rr, cc = rectangle_perimeter(start, extent=extent, shape=img.shape,
|
||
|
|
clip=False)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# clip=True
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1],
|
||
|
|
[0, 1, 0, 0, 1],
|
||
|
|
[0, 1, 0, 0, 1],
|
||
|
|
[0, 1, 1, 1, 1]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(start, extent=extent, shape=img.shape,
|
||
|
|
clip=True)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_perimiter_clip_top_left():
|
||
|
|
# clip=False
|
||
|
|
expected = np.array([[0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 1, 0],
|
||
|
|
[0, 0, 0, 1, 0],
|
||
|
|
[1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
start = (-5, -5)
|
||
|
|
end = (2, 2)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=False)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
# clip=True
|
||
|
|
expected = np.array([[1, 1, 1, 1, 0],
|
||
|
|
[1, 0, 0, 1, 0],
|
||
|
|
[1, 0, 0, 1, 0],
|
||
|
|
[1, 1, 1, 1, 0],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=True)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_perimiter_clip_top_right():
|
||
|
|
expected = np.array([[0, 1, 1, 1, 1],
|
||
|
|
[0, 1, 0, 0, 1],
|
||
|
|
[0, 1, 0, 0, 1],
|
||
|
|
[0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
start = (-10, 2)
|
||
|
|
end = (2, 10)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=True)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
expected = np.array([[0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 0, 0, 0],
|
||
|
|
[0, 1, 1, 1, 1],
|
||
|
|
[0, 0, 0, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=False)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_rectangle_perimiter_clip_bottom_left():
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0],
|
||
|
|
[1, 1, 1, 0, 0],
|
||
|
|
[1, 0, 1, 0, 0],
|
||
|
|
[1, 0, 1, 0, 0],
|
||
|
|
[1, 1, 1, 0, 0]], dtype=np.uint8)
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
start = (2, -3)
|
||
|
|
end = (10, 1)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=True)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|
||
|
|
|
||
|
|
expected = np.array([[0, 0, 0, 0, 0],
|
||
|
|
[1, 1, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0],
|
||
|
|
[0, 0, 1, 0, 0]], dtype=np.uint8)
|
||
|
|
|
||
|
|
img = np.zeros((5, 5), dtype=np.uint8)
|
||
|
|
rr, cc = rectangle_perimeter(start, end=end, shape=img.shape,
|
||
|
|
clip=False)
|
||
|
|
img[rr, cc] = 1
|
||
|
|
assert_array_equal(img, expected)
|