def test_proposal_infer_one_output(self):
graph = build_graph(nodes_attributes,
[('proposal_input', 'proposal'),
('proposal', 'proposal_out_data_1'),
('proposal_out_data_1', 'op_output')
],
{'proposal_input': {'shape': int64_array([1, 3, 227, 227])},
'proposal': {'post_nms_topn': 2, **layout_attrs()}
})
proposal_node = Node(graph, 'proposal')
ProposalOp.proposal_infer(proposal_node)
self.assertListEqual([1 * 2, 5], list(graph.node['proposal_out_data_1']['shape']))
def test_proposal_infer(self):
graph = build_graph(nodes_attributes,
[('node_1', 'proposal'),
('proposal', 'node_3')],
{'node_3': {'is_output': True, 'shape': None},
'node_1': {'shape': np.array([1, 3, 227, 227])},
'proposal': {'post_nms_topn': 2, **layout_attrs()}
})
proposal_node = Node(graph, 'proposal')
ProposalOp.proposal_infer(proposal_node)
exp_shape = np.array([1 * 2, 5])
res_shape = graph.node['node_3']['shape']
for i in range(0, len(exp_shape)):
self.assertEqual(exp_shape[i], res_shape[i])