Updates the graph to replace all ResizeNearest ops with ResizeNearest plugins in backbone.
()
| 352 | :param second_nms_threshold: Override the 2nd NMS score threshold value. If set to None, use the value in the graph. |
| 353 | """ |
| 354 | def backbone(): |
| 355 | """ |
| 356 | Updates the graph to replace all ResizeNearest ops with ResizeNearest plugins in backbone. |
| 357 | """ |
| 358 | # Get final backbone outputs. |
| 359 | p2 = self.graph.find_node_by_op_name("Conv", "/backbone/fpn_output2/Conv") |
| 360 | p3 = self.graph.find_node_by_op_name("Conv", "/backbone/fpn_output3/Conv") |
| 361 | p4 = self.graph.find_node_by_op_name("Conv", "/backbone/fpn_output4/Conv") |
| 362 | p5 = self.graph.find_node_by_op_name("Conv", "/backbone/fpn_output5/Conv") |
| 363 | |
| 364 | |
| 365 | return p2.outputs[0], p3.outputs[0], p4.outputs[0], p5.outputs[0] |
| 366 | |
| 367 | def proposal_generator(anchors, first_nms_threshold): |
| 368 | """ |