model-optimizer/extensions/front/tf/legacy_mask_rcnn_support.json

   1 [
   2      {
   3         "custom_attributes": {
   4         },
   5         "id": "PreprocessorReplacement",
   6         "inputs": [
   7             [
   8                 {
   9                     "node": "map/Shape$",
  10                     "port": 0
  11                 },
  12                 {
  13                     "node": "map/TensorArrayUnstack/Shape$",
  14                     "port": 0
  15                 },
  16                 {
  17                     "node": "map/TensorArrayUnstack/TensorArrayScatter/TensorArrayScatterV3$",
  18                     "port": 2
  19                 }
  20             ]
  21         ],
  22         "instances": [
  23             ".*Preprocessor/"
  24         ],
  25         "match_kind": "scope",
  26         "outputs": [
  27             {
  28                 "node": "sub$",
  29                 "port": 0
  30             },
  31             {
  32                 "node": "map/TensorArrayStack_1/TensorArrayGatherV3$",
  33                 "port": 0
  34             }
  35         ]
  36     },
  37     {
  38         "custom_attributes": {
  39             "input_height": 800,
  40             "input_width": 800,
  41             "nms_threshold": 0.7,
  42             "feat_stride": 16,
  43             "max_proposals": 100,
  44             "anchor_base_size": 256,
  45             "anchor_scales": [0.25, 0.5, 1.0, 2.0],
  46             "anchor_aspect_ratios": [0.5, 1.0, 2.0],
  47             "roi_spatial_scale": 0.0625
  48         },
  49         "id": "TFObjectDetectionAPIMaskRCNNProposalAndROIPooling",
  50         "include_inputs_to_sub_graph": true,
  51         "include_outputs_to_sub_graph": true,
  52         "instances": {
  53             "end_points": [
  54                 "CropAndResize",
  55                 "map_1/TensorArrayStack/TensorArrayGatherV3",
  56                 "BatchMultiClassNonMaxSuppression/map/TensorArrayStack_4/TensorArrayGatherV3",
  57                 "map_1/Shape"
  58             ],
  59             "start_points": [
  60                 "FirstStageBoxPredictor/concat",
  61                 "FirstStageBoxPredictor/concat_1",
  62                 "GridAnchorGenerator/Identity",
  63                 "Shape",
  64                 "CropAndResize"
  65             ]
  66         },
  67         "match_kind": "points"
  68     },
  69     {
  70        "custom_attributes": {
  71             "code_type": "caffe.PriorBoxParameter.CENTER_SIZE",
  72             "confidence_threshold": 0.01,
  73             "keep_top_k": 100,
  74             "input_height": 800,
  75             "input_width": 800,
  76             "nms_threshold": 0.6,
  77             "pad_mode": "caffe.ResizeParameter.CONSTANT",
  78             "resize_mode": "caffe.ResizeParameter.WARP",
  79             "max_detections_per_class": 100,
  80             "num_classes": 90
  81         },
  82         "id": "TFObjectDetectionAPIMaskRCNNDetectionOutput",
  83         "include_inputs_to_sub_graph": true,
  84         "include_outputs_to_sub_graph": true,
  85         "instances": {
  86             "end_points": [
  87                 "BatchMultiClassNonMaxSuppression_1/map/TensorArrayStack_2/TensorArrayGatherV3",
  88                 "BatchMultiClassNonMaxSuppression_1/map/TensorArrayStack/TensorArrayGatherV3"
  89             ],
  90             "start_points": [
  91                 "SecondStageBoxPredictor/Reshape",
  92                 "SecondStageBoxPredictor/Reshape_1",
  93                 "ExpandDims_7",
  94                 "ToFloat_6"
  95             ]
  96         },
  97         "match_kind": "points"
  98     },
  99     {
 100         "custom_attributes": {
 101             "roi_spatial_scale": 1.0
 102         },
 103         "id": "TFObjectDetectionAPIMaskRCNNROIPoolingSecond",
 104         "include_inputs_to_sub_graph": true,
 105         "include_outputs_to_sub_graph": true,
 106         "instances": {
 107             "end_points": [
 108                 "CropAndResize_1"
 109             ],
 110             "start_points": [
 111                 "CropAndResize_1"
 112             ]
 113         },
 114         "match_kind": "points"
 115     },
 116     {
 117        "custom_attributes": {
 118         },
 119         "id": "TFObjectDetectionAPIMaskRCNNSigmoid",
 120         "include_inputs_to_sub_graph": true,
 121         "include_outputs_to_sub_graph": true,
 122         "instances": {
 123             "end_points": [
 124                 "SecondStageBoxPredictor_1/Conv_1/BiasAdd"
 125             ],
 126             "start_points": [
 127                 "SecondStageBoxPredictor_1/Conv_1/BiasAdd"
 128             ]
 129         },
 130         "match_kind": "points"
 131     }
 132 ]