def test_bbox_transform(self):
# TODO: test correct-ness
sizes = [12]
strides = [8]
ratios = np.array([1, 2], K.floatx())
scales = np.array([1, 2], K.floatx())
anchor_params = utils.AnchorParameters(sizes, strides, ratios, scales)
pyramid_levels = [3]
image_shape = (16, 16)
anchors = utils.anchors_for_shape(image_shape,
pyramid_levels=pyramid_levels,
anchor_params=anchor_params)
# test custom std/mean
targets = utils.bbox_transform(anchors,
np.random.random((1, 4)),
mean=[0],
std=[0.2])
self.assertTupleEqual(targets.shape, (16, 4))
# test bad `mean` value
with self.assertRaises(ValueError):
utils.bbox_transform(anchors, [1], mean='invalid', std=None)
# test image / annotation not empty
with self.assertRaises(ValueError):
utils.bbox_transform(anchors, [1], mean=None, std='invalid')
def test_anchor_targets_bbox(self):
# TODO: test correct-ness
sizes = [12]
strides = [8]
ratios = np.array([1, 2], K.floatx())
scales = np.array([1, 2], K.floatx())
anchor_params = utils.AnchorParameters(sizes, strides, ratios, scales)
pyramid_levels = [3]
image_shape = (16, 16)
anchors = utils.anchors_for_shape(image_shape,
pyramid_levels=pyramid_levels,
anchor_params=anchor_params)
# test image / annotation size mismatch
with self.assertRaises(ValueError):
utils.anchor_targets_bbox(anchors, [1], [1, 2, 3], 1)
# test image / annotation not empty
with self.assertRaises(ValueError):
utils.anchor_targets_bbox(anchors, [], [], 1)
# test annotation structure
with self.assertRaises(ValueError):
utils.anchor_targets_bbox(anchors, [1], [{'labels': 1}], 1)
with self.assertRaises(ValueError):
utils.anchor_targets_bbox(anchors, [1], [{'bboxes': 1}], 1)
def test_anchors_for_shape_dimensions(self):
sizes = [32, 64, 128]
strides = [8, 16, 32]
ratios = np.array([0.5, 1, 2, 3], K.floatx())
scales = np.array([1, 1.2, 1.6], K.floatx())
anchor_params = utils.AnchorParameters(sizes, strides, ratios, scales)
pyramid_levels = [3, 4, 5]
image_shape = tensor_shape.TensorShape((64, 64))
all_anchors = utils.anchors_for_shape(image_shape,
pyramid_levels=pyramid_levels,
anchor_params=anchor_params)
self.assertTupleEqual(all_anchors.shape, (1008, 4))
self.assertEqual(anchor_params.num_anchors(), 12)
def test_anchors_for_shape_values(self):
sizes = [12]
strides = [8]
ratios = np.array([1, 2], K.floatx())
scales = np.array([1, 2], K.floatx())
anchor_params = utils.AnchorParameters(sizes, strides, ratios, scales)
pyramid_levels = [3]
image_shape = (16, 16)
all_anchors = utils.anchors_for_shape(image_shape,
pyramid_levels=pyramid_levels,
anchor_params=anchor_params)
# using almost_equal for floating point imprecisions
self.assertAllClose(all_anchors[0, :], [
strides[0] / 2 - (sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 - (sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[1, :], [
strides[0] / 2 - (sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 - (sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[2, :], [
strides[0] / 2 - (sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 - (sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[3, :], [
strides[0] / 2 - (sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 - (sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[4, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 - (sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[5, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 - (sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[6, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 - (sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[7, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 - (sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[8, :], [
strides[0] / 2 - (sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[9, :], [
strides[0] / 2 - (sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[10, :], [
strides[0] / 2 - (sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[11, :], [
strides[0] / 2 - (sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
strides[0] / 2 + (sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[12, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[13, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] / np.sqrt(ratios[0])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] * np.sqrt(ratios[0])) / 2,
])
self.assertAllClose(all_anchors[14, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[0] * np.sqrt(ratios[1])) / 2,
])
self.assertAllClose(all_anchors[15, :], [
strides[0] * 3 / 2 -
(sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 -
(sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] / np.sqrt(ratios[1])) / 2,
strides[0] * 3 / 2 +
(sizes[0] * scales[1] * np.sqrt(ratios[1])) / 2,
])