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

   1 [
   2     {
   3         "custom_attributes": {
   4             "code_type": "caffe.PriorBoxParameter.CORNER",
   5             "pad_mode": "caffe.ResizeParameter.CONSTANT",
   6             "resize_mode": "caffe.ResizeParameter.WARP",
   7             "confidence_threshold": 0.05,
   8             "top_k": 6000,
   9             "keep_top_k": 300,
  10             "nms_threshold": 0.5,
  11             "variance": [0.2, 0.2, 0.2, 0.2]
  12         },
  13         "include_inputs_to_sub_graph": true,
  14         "include_outputs_to_sub_graph": true,
  15         "id": "RetinaNetFilteredDetectionsReplacement",
  16         "instances": {
  17             "end_points": [
  18                 "filtered_detections/map/TensorArrayStack/TensorArrayGatherV3",
  19                 "filtered_detections/map/TensorArrayStack_1/TensorArrayGatherV3",
  20                 "filtered_detections/map/TensorArrayStack_2/TensorArrayGatherV3"
  21             ],
  22             "start_points": [
  23                 "regression/concat",
  24                 "classification/concat",
  25                 "anchors/concat",
  26                 "clipped_boxes/Shape"
  27             ]
  28         },
  29         "match_kind": "points"
  30     }
  31 ]