def test_bifpn_l3l7(self):
p1 = fpn_configs.bifpn_config(3, 7, None)
# pyformat: disable
self.assertEqual(
p1.nodes,
[
{'feat_level': 6, 'inputs_offsets': [3, 4]},
{'feat_level': 5, 'inputs_offsets': [2, 5]},
{'feat_level': 4, 'inputs_offsets': [1, 6]},
{'feat_level': 3, 'inputs_offsets': [0, 7]},
{'feat_level': 4, 'inputs_offsets': [1, 7, 8]},
{'feat_level': 5, 'inputs_offsets': [2, 6, 9]},
{'feat_level': 6, 'inputs_offsets': [3, 5, 10]},
{'feat_level': 7, 'inputs_offsets': [4, 11]},
])
def test_bifpn_l2l7(self):
p = fpn_configs.bifpn_config(2, 7, None)
# pyformat: disable
self.assertEqual(p.nodes, [
{
'feat_level': 6,
'inputs_offsets': [4, 5]
},
{
'feat_level': 5,
'inputs_offsets': [3, 6]
},
{
'feat_level': 4,
'inputs_offsets': [2, 7]
},
{
'feat_level': 3,
'inputs_offsets': [1, 8]
},
{
'feat_level': 2,
'inputs_offsets': [0, 9]
},
{
'feat_level': 3,
'inputs_offsets': [1, 9, 10]
},
{
'feat_level': 4,
'inputs_offsets': [2, 8, 11]
},
{
'feat_level': 5,
'inputs_offsets': [3, 7, 12]
},
{
'feat_level': 6,
'inputs_offsets': [4, 6, 13]
},
{
'feat_level': 7,
'inputs_offsets': [5, 14]
},
])