def test_compute_bboxes_from_scoremaps_degenerate(self):
scoremap = np.zeros([3, 3], dtype=np.float)
scoremap_threshold_list = np.arange(0, 1, 0.2)
boxes = compute_bboxes_from_scoremaps(scoremap,
scoremap_threshold_list)
self.assertListEqual(boxes, [[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0],
[0, 0, 0, 0], [0, 0, 0, 0]])
def test_compute_bboxes_from_scoremaps_multimodal(self):
# array has shape array[y][x]
scoremap = np.array(
[[0.4, 0.0, 0.2, 0.2, 0.2], [0.4, 0.4, 0.0, 0.4, 0.0],
[0.0, 0.0, 0.0, 0.4, 0.0], [1.0, 0.6, 0.8, 0.2, 0.2]],
dtype=np.float)
scoremap_threshold_list = np.arange(0, 1, 0.2)
boxes = compute_bboxes_from_scoremaps(scoremap,
scoremap_threshold_list)
self.assertListEqual(boxes, [[0, 0, 4, 3], [0, 0, 2, 2], [0, 3, 3, 3],
[2, 3, 3, 3], [0, 3, 1, 3]])
def test_compute_bboxes_from_scoremaps_unimodal(self):
# array has shape array[y][x]
scoremap = np.array(
[[0.0, 0.0, 0.0, 0.0, 0.0], [0.0, 0.4, 0.4, 0.4, 0.6],
[0.0, 0.4, 1.0, 0.8, 0.6], [0.0, 0.4, 0.4, 0.4, 0.6]],
dtype=np.float)
scoremap_threshold_list = np.arange(0, 1, 0.2)
boxes = compute_bboxes_from_scoremaps(scoremap,
scoremap_threshold_list)[0]
boxes = [box[0].tolist() for box in boxes]
self.assertListEqual(boxes, [[1, 1, 4, 3], [1, 1, 4, 3], [2, 1, 4, 3],
[2, 2, 4, 3], [2, 2, 3, 3]])