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] }, ])