Exemplo n.º 1
0
 def test_anchor_match(self):
   anchors = [(2.0, 2.0), (1.0, 1.0), (1.5, 1.0)]
   _, __, anchor_idx1 = get_box_grid_position((0, 0, 0.1, 0.1), 10, anchors)
   _, __, anchor_idx2 = get_box_grid_position((0, 0, 0.2, 0.2), 10, anchors)
   _, __, anchor_idx3 = get_box_grid_position((0, 0, 0.15, 0.1), 10, anchors)
   self.assertEqual(1, anchor_idx1)
   self.assertEqual(0, anchor_idx2)
   self.assertEqual(2, anchor_idx3)
Exemplo n.º 2
0
 def test_grid_position_0_0(self):
   col, row, anchor_idx = get_box_grid_position((0, 0, 0.1, 0.1), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(0, col)
   self.assertEqual(0, row)
Exemplo n.º 3
0
 def test_grid_position_exceeds_positive_9_9(self):
   col, row, anchor_idx = get_box_grid_position((1.5, 1.5, 0.1, 0.1), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(9, col)
   self.assertEqual(9, row)
Exemplo n.º 4
0
 def test_grid_position_exceeds_negative_0_0(self):
   col, row, anchor_idx = get_box_grid_position((-0.5, -0.5, 0.1, 0.1), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(0, col)
   self.assertEqual(0, row)
Exemplo n.º 5
0
 def test_grid_position_ct_3_5(self):
   col, row, anchor_idx = get_box_grid_position((0.2, 0.4, 0.2, 0.2), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(3, col)
   self.assertEqual(5, row)
Exemplo n.º 6
0
 def test_grid_position_ct_1_1(self):
   col, row, anchor_idx = get_box_grid_position((0, 0, 0.2, 0.2), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(1, col)
   self.assertEqual(1, row)
Exemplo n.º 7
0
 def test_grid_position_2_4(self):
   col, row, anchor_idx = get_box_grid_position((0.2, 0.4, 0.1, 0.1), 10, [(1.0, 1.0)])
   self.assertEqual(0, anchor_idx)
   self.assertEqual(2, col)
   self.assertEqual(4, row)