def test_oriented_bbox_one_outside_high_threshold(self):
bbox = np.array((
(2, 3),
(8, 1),
(9, 4),
(3, 6),
), dtype='float32')
bbox[:,1] -= 2.0
self.assertIsNone(OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.7))
def test_orthogonal_bbox_all_corners_inside(self):
bbox = np.array((
(3, 3),
(5, 3),
(5, 7),
(3, 7),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, bbox)
def test_orthogonal_bbox_corners_inside_but_touching_corner_of_boundary(self):
bbox = np.array((
(0, 3),
(5, 3),
(5, 10),
(0, 10),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, bbox)
def test_failed_bbox_crops_4(self):
crop_shape = (1400, 1400)
bbox = np.array([
[66.9483, 5.626648],
[65.91528, -0.5026245],
[-62.482483, 21.137451],
[-61.449463, 27.266724]
], dtype='float32')
crop = OBBox.crop_bbox(bbox, crop_shape)
self.assertTrue(OBBox.is_bbox_inside(crop, crop_shape))
def test_failed_bbox_crops_3(self):
crop_shape = (1400, 1400)
bbox = np.array([
[50.13446, 126.713806],
[-3.972412, 123.53104],
[-4.3586426, 130.09656],
[49.74829, 133.27933]
], dtype='float32')
crop = OBBox.crop_bbox(bbox, crop_shape)
self.assertTrue(OBBox.is_bbox_inside(crop, crop_shape))
def test_failed_bbox_crops_2(self):
crop_shape = (1400, 1400)
bbox = np.array((
(319.46906, -2.0354004),
(304.40265, -3.0398254),
(301.16815, 45.477875 ),
(316.2345, 46.4823 )
), dtype='float32')
crop = OBBox.crop_bbox(bbox, crop_shape)
self.assertTrue(OBBox.is_bbox_inside(crop, crop_shape))
def test_failed_bbox_crops_1(self):
crop_shape = (1400, 1400)
bbox = np.array((
(1552.2725, 111.535095),
(1550.8269, 105.246704),
(1320.6572, 158.15924),
(1322.1029, 164.44763),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, crop_shape)
self.assertTrue(OBBox.is_bbox_inside(crop, crop_shape))
def test_oriented_bbox_all_corners_inside(self):
bbox = np.array((
(2, 3),
(8, 1),
(9, 4),
(3, 6),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, bbox)
def test_oriented_bbox_four_outside(self):
bbox = np.array((
(-2, 7),
(11, 4),
(12, 7),
(-1, 10),
), dtype='float32')
new_bbox = np.array((
(0, 6.5384615),
(9, 4.4615384),
(10, 7.4615384),
(1, 9.5384615),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.5)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, new_bbox)
def test_oriented_bbox_one_outside(self):
bbox = np.array((
(2, 3),
(8, 1),
(9, 4),
(3, 6),
), dtype='float32')
bbox[:,1] -= 2.0
new_bbox = np.array((
(2.3333333, 2),
(8.3333333, 0),
(9, 2),
(3, 4),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.6)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, new_bbox)
def test_oriented_bbox_three_outside(self):
bbox = np.array((
(2, 3),
(8, 1),
(9, 4),
(3, 6),
), dtype='float32')
bbox[:,1] -= 4.0
bbox[:,0] += 2.0
new_bbox = np.array((
(4.666667, 1.),
(7.6666665, 0.),
(8., 1. ),
(5., 2.)
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.0)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, new_bbox)
def test_oriented_bbox_two_outside_one_touching(self):
bbox = np.array((
(2, 3),
(8, 1),
(9, 4),
(3, 6),
), dtype='float32')
bbox[:,1] -= 3.0
bbox[:,0] += 2.0
new_bbox = np.array((
(4.5555555, 1.6666666),
(9.5555555, 0),
(10, 1.3333333),
(5, 3),
), dtype='float32')
crop = OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.3)
self.assertTrue(OBBox.is_bbox_inside(crop, self.BOUNDARY))
np.testing.assert_almost_equal(crop, new_bbox)