def test_bifpn_dynamic_l2l7(self):
p = efficientdet_arch.bifpn_dynamic_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]
},
])
def test_bifpn_dynamic_l3l7(self):
p1 = efficientdet_arch.bifpn_dynamic_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_dynamic_l3l7(self):
p1 = efficientdet_arch.bifpn_dynamic_config(3, 7, None)
p2 = efficientdet_arch.bifpn_fa_config()
self.assertEqual(p1.weight_method, p2.weight_method)
self.assertEqual(p1.nodes, p2.nodes)
