4547818fb1
* Refactored code, updated comments and documentation related to TF OD API models pre-processing. * Improved MO messages related to pre-processor block removal during conversion of the TD OD API models. Remove mean/scale if padding is used and mean/scale is applied before resize * Updated TF OD API transformation and documentation for SSD models * Updated comments and documentation for the ObjectDetectionAPIMaskRCNNSigmoidReplacement transformation * Updated comments and documentation for the ObjectDetectionAPIMaskRCNNROIPoolingSecondReplacement transformation * Updated comments and documentation for the ObjectDetectionAPIPSROIPoolingReplacement transformation * Updated comments and documentation for the ObjectDetectionAPIProposalReplacement transformation * Updated comments and documentation for the ObjectDetectionAPIDetectionOutputReplacement transformation * Minor code style fixes * Fixed unit tests for ObjectDetectionAPIPreprocessor2Replacement transformation * Improved unit test for pipeline.config parser. Fixed very long bug with incorrect test data for the PipelineConfig parser class * Code style fixes * Get rid of "coordinates_swap_method" parameter in the JSON configuration file for TF OD API models * Code style fixes and minor refactoring * Simplied code related to swapping Proposal coordinates * Removed incorrectly removed code * Fixed code review comments about the code comments
113 lines
3.0 KiB
JSON
113 lines
3.0 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
|
|
},
|
|
"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"
|
|
}
|
|
]
|