Evgeny Lazarev
c6bfac6e05
* Config for TF 2.0 Faster R-CNN models, refactored subgraph_between_nodes to use graph API * Added support for new type of Preprocessing block in the TF 2.0 OD API models. Various fixes to enable the Faster R-CNN ResNet 50 * Updated text comments * Fixed sub_graph_between_nodes for TensorIteratorMerge. Added support for the TF 2.X EfficientDet models (not yet reshape-able) * Fixed unit tests * Fixed regression for TF 1.X OD API SSD model, enabled TF 2.0 OD API SSD models * Code clean up * Switched TF 2.0 OD API Faster R-CNN to preprocessor replacement type 2 * Refactored ObjectDetectionAPIPreprocessorReplacement and ObjectDetectionAPIPreprocessor2Replacement * Fixed bug in the Div transformation to Mul when input is integer. * Added support for the TF 2.0 OD API Mask R-CNN * Added unit tests for Div operation. Updated incorrectly modified mask_rcnn_support_api_v1.14.json * Updated document with list of supported configuration files for TF OD API models * Review comments * Added tests for control flow edges for the sub_graph_between_nodes function * Two more tests
51 lines
2.1 KiB
JSON
51 lines
2.1 KiB
JSON
[
|
|
{
|
|
"custom_attributes": {
|
|
"start_nodes": ["StatefulPartitionedCall/Preprocessor/unstack"],
|
|
"end_nodes": ["StatefulPartitionedCall/Preprocessor/stack",
|
|
"StatefulPartitionedCall/Preprocessor/stack_1"]
|
|
},
|
|
"id": "ObjectDetectionAPIPreprocessor2Replacement",
|
|
"match_kind": "general"
|
|
},
|
|
{
|
|
"custom_attributes": {
|
|
"code_type": "caffe.PriorBoxParameter.CENTER_SIZE",
|
|
"pad_mode": "caffe.ResizeParameter.CONSTANT",
|
|
"resize_mode": "caffe.ResizeParameter.WARP",
|
|
"clip_before_nms": false,
|
|
"clip_after_nms": true,
|
|
"disable_prior_boxes_layers_generator": true
|
|
},
|
|
"id": "ObjectDetectionAPISSDPostprocessorReplacement",
|
|
"include_inputs_to_sub_graph": true,
|
|
"include_outputs_to_sub_graph": true,
|
|
"instances": {
|
|
"end_points": [
|
|
"StatefulPartitionedCall/Identity",
|
|
"StatefulPartitionedCall/Identity_1",
|
|
"StatefulPartitionedCall/Identity_2",
|
|
"StatefulPartitionedCall/Identity_3",
|
|
"StatefulPartitionedCall/Identity_4",
|
|
"StatefulPartitionedCall/Identity_5",
|
|
"StatefulPartitionedCall/Identity_6",
|
|
"StatefulPartitionedCall/Identity_7"
|
|
],
|
|
"start_points": [
|
|
"StatefulPartitionedCall/Postprocessor/Reshape_1",
|
|
"StatefulPartitionedCall/Postprocessor/scale_logits",
|
|
"StatefulPartitionedCall/Postprocessor/Tile",
|
|
"StatefulPartitionedCall/Postprocessor/Cast_1"
|
|
]
|
|
},
|
|
"match_kind": "points"
|
|
},
|
|
{
|
|
"custom_attributes": {
|
|
"outputs": "StatefulPartitionedCall/Identity,StatefulPartitionedCall/Identity_1,StatefulPartitionedCall/Identity_2,StatefulPartitionedCall/Identity_3,StatefulPartitionedCall/Identity_4,StatefulPartitionedCall/Identity_5,StatefulPartitionedCall/Identity_6,StatefulPartitionedCall/Identity_7"
|
|
},
|
|
"id": "ObjectDetectionAPIOutputReplacement",
|
|
"match_kind": "general"
|
|
}
|
|
]
|