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_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_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_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_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_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_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_orthogonal_bbox_four_corners_out(self):
bbox = np.array((
(-1, 2),
(11, 2),
(11, 7),
(-1, 7),
), dtype='float32')
new_bbox = np.array((
(0, 2),
(10, 2),
(10, 7),
(0, 7),
), 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_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_orthogonal_bbox_three_corners_out_high_threshold(self):
bbox = np.array((
(3, -2),
(11, -2),
(11, 7),
(3, 7),
), dtype='float32')
self.assertIsNone(OBBox.crop_bbox(bbox, self.BOUNDARY, threshold_rel=0.7, threshold_abs=100.0))
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)
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))
# x1, y1 = ImageFont.load_default().getsize(text)
# x1 += position[0] + 4
# y1 += position[1] + 4
# draw.rectangle((position[0], position[1], x1, y1), fill=color_box)
# draw.text((position[0] + 2, position[1] + 2), text, color_text)
# return draw, x1, position[1]
STAT_FILE = 'stats.json'
if args.compile:
stats = {}
areas_by_cat = {}
print("Gathering the stats")
for cat_id, o_bbox in zip(dataset.obb.ann_info['cat_id'],
dataset.obb.ann_info['o_bbox']):
cat = int(cat_id[0])
bbox = np.array(o_bbox).reshape((4, 2))
area = OBBox.get_area(bbox)
areas_by_cat[cat] = np.append(areas_by_cat.get(cat, np.array([])),
area)
print("Calculating the stats")
for cat, areas in areas_by_cat.items():
# cat = int(cat_id[0])
stats[cat] = {
'name': dataset.cat2label[cat],
'min': min(areas),
'max': max(areas),
'mean': np.mean(areas),
'median': np.median(areas),
'std': np.std(areas),
'length': areas.size,
'sorted_areas': sorted(areas, reverse=True)
}