Files
openvino/model-optimizer/extensions/front/tf/faster_rcnn_support.json
T

114 lines
3.1 KiB
JSON

[
{
"custom_attributes": {
},
"id": "ObjectDetectionAPIPreprocessorReplacement",
"inputs": [
[
{
"node": "map/Shape$",
"port": 0
},
{
"node": "map/TensorArrayUnstack/Shape$",
"port": 0
},
{
"node": "map/TensorArrayUnstack/TensorArrayScatter/TensorArrayScatterV3$",
"port": 2
}
]
],
"instances": [
".*Preprocessor/"
],
"match_kind": "scope",
"outputs": [
{
"node": "sub$",
"port": 0
},
{
"node": "map/TensorArrayStack_1/TensorArrayGatherV3$",
"port": 0
}
]
},
{
"custom_attributes": {
"clip_before_nms": true,
"clip_after_nms": false
},
"id": "ObjectDetectionAPIProposalReplacement",
"include_inputs_to_sub_graph": true,
"include_outputs_to_sub_graph": true,
"instances": {
"end_points": [
"map/TensorArrayStack/TensorArrayGatherV3",
"map_1/TensorArrayStack/TensorArrayGatherV3",
"BatchMultiClassNonMaxSuppression/map/TensorArrayStack_4/TensorArrayGatherV3"
],
"start_points": [
"FirstStageBoxPredictor/Reshape",
"FirstStageBoxPredictor/Reshape_1",
"GridAnchorGenerator/Identity",
"Shape"
]
},
"match_kind": "points"
},
{
"custom_attributes": {
"clip_before_nms": true,
"clip_after_nms": false,
"coordinates_swap_method": "swap_weights"
},
"id": "ObjectDetectionAPIDetectionOutputReplacement",
"inputs": [
[
{
"node": "Reshape$",
"port": 0
}
],
[
{
"node": "Reshape_1$",
"port": 0
}
],
[
{
"node": "ExpandDims$",
"port": 0
}
]
],
"instances": [
".*SecondStagePostprocessor/"
],
"match_kind": "scope",
"outputs": [
{
"node": "BatchMultiClassNonMaxSuppression/map/TensorArrayStack/TensorArrayGatherV3$",
"port": 0
}
]
},
{
"custom_attributes": {
"outputs": "detection_boxes,detection_scores,num_detections"
},
"id": "ObjectDetectionAPIOutputReplacement",
"match_kind": "general"
},
{
"custom_attributes":
{
"replacements": [["mul/y", "first_stage_max_proposals"]]
},
"id": "ObjectDetectionAPIConstValueOverride",
"match_kind": "general"
}
]