def test_rgb_to_hsv(self):
image = np.array(
[[[1, 2, 3], [0, 0, 0]],
[[0, 0, 0], [1, 1, 1]]])
expected = np.array(
[[[149, 0], [0, 0]],
[[170, 0], [0, 0]],
[[3, 0], [0, 1]]])
actual = rgb_to_hsv(image)
assert_array_equal(actual, expected)
def _get_colour(self, numpy_array):
hue, sat, val = rgb_to_hsv(numpy_array)
hmin = hue.min()
hmax = hue.max()
smin = sat.min()
smax = sat.max()
vmin = val.min()
vmax = val.max()
print(hmin, hmax, smin, smax, vmin, vmax)
return hsv(lambda h, s, v: ((h >= hmin) & (h <= hmax)) &
((s >= smin) & (s <= smax)) & ((v >= vmin) & (v <= vmax)))
def _get_colour(self, numpy_array):
hue, sat, val = rgb_to_hsv(numpy_array)
hmin = hue.min()
hmax = hue.max()
smin = sat.min()
smax = sat.max()
vmin = val.min()
vmax = val.max()
return hsv(lambda h, s, v: (
(h >= hmin) & (h <= hmax)) &
((s >= smin) & (s <= smax)) &
((v >= vmin) & (v <= vmax)))
def test_rgb_to_hsv(self):
image = np.array([[[1, 2, 3], [0, 0, 0]], [[0, 0, 0], [1, 1, 1]]])
expected = np.array([[[149, 0], [0, 0]], [[170, 0], [0, 0]],
[[3, 0], [0, 1]]])
actual = rgb_to_hsv(image)
assert_array_equal(actual, expected)
