| 441 | } |
| 442 | |
| 443 | NodeID GraphBuilder::add_detection_output_node(Graph &g, |
| 444 | NodeParams params, |
| 445 | NodeIdxPair input_loc, |
| 446 | NodeIdxPair input_conf, |
| 447 | NodeIdxPair input_priorbox, |
| 448 | const DetectionOutputLayerInfo &detect_info) |
| 449 | { |
| 450 | check_nodeidx_pair(input_loc, g); |
| 451 | check_nodeidx_pair(input_conf, g); |
| 452 | check_nodeidx_pair(input_priorbox, g); |
| 453 | |
| 454 | // Create detection_output node and connect |
| 455 | NodeID detect_nid = g.add_node<DetectionOutputLayerNode>(detect_info); |
| 456 | g.add_connection(input_loc.node_id, input_loc.index, detect_nid, 0); |
| 457 | g.add_connection(input_conf.node_id, input_conf.index, detect_nid, 1); |
| 458 | g.add_connection(input_priorbox.node_id, input_priorbox.index, detect_nid, 2); |
| 459 | |
| 460 | set_node_params(g, detect_nid, params); |
| 461 | |
| 462 | return detect_nid; |
| 463 | } |
| 464 | |
| 465 | NodeID GraphBuilder::add_detection_post_process_node(Graph &g, |
| 466 | NodeParams params, |
nothing calls this directly
no test coverage detected