def test_with_no_scores_field(self):
boxlist = np_box_list_ops.copy_boxlist(boxlist=self.boxlist)
max_output_size = 3
iou_threshold = 0.5
with self.assertRaises(ValueError):
np_box_list_ops.non_max_suppression3d(boxlist, max_output_size,
iou_threshold)
def test_nms(self):
boxlist = self.boxlist
max_output_size = 3
iou_threshold = 0.1
expected_boxes = np.array(
[[_degree_to_radians(20.0), 1.0, 1.0, 1.0, 10.0, 0.0, 0.0],
[_degree_to_radians(-10.0), 1.0, 1.0, 1.0, 0.0, 0.0, 0.0]],
dtype=float)
nms_boxlist = np_box_list_ops.non_max_suppression3d(
boxlist, max_output_size, iou_threshold)
self._compare_boxlist_with_boxes(nms_boxlist, expected_boxes)