Evgeny Lazarev
522ad39a48
* Do not run transformations for TF 2.X OD API models recursively (needed for models with Loop operation) * Added anchor front transformation to group all TF OD API transformations. Added new necessary dependencies from KerasRNN transformations related to While support * Added JSON configuration files for TF 2.4 OD API SSD and EfficientDet models * Updated documentation with table of supported TF 2.x OD API models * Improved visualization of the dependency graph * Updated version of the pre-processing transformation for TF 2.4 OD API models * Fixes in the TF 2.x OD API models conversion * Fixed order of applying mean/scale values for TF 2.X OD API pre-processing * Updates to the documentation * Fixes for the preprocessor block transformation for the TF OD API models * Added code comments * Fixed bom file * Unit tests for the TF 2.4 OD API ObjectDetectionAPIPreprocessor2Replacement transformation * Code cleanup * Updates to the documentation on how to convert TF OD API models and graph dumper * Added assert to make sure that operations in the `get_specific_ops_with_const_inputs` has exactly 2 inputs
51 lines
2.2 KiB
JSON
51 lines
2.2 KiB
JSON
[
|
|
{
|
|
"custom_attributes": {
|
|
"start_nodes": ["StatefulPartitionedCall/map/TensorArrayUnstack/TensorListFromTensor"],
|
|
"end_nodes": ["StatefulPartitionedCall/map/TensorArrayV2Stack/TensorListStack",
|
|
"StatefulPartitionedCall/map/TensorArrayV2Stack_1/TensorListStack"]
|
|
},
|
|
"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"
|
|
}
|
|
]
|