diff options
author | keidav01 <keith.davis@arm.com> | 2019-02-26 11:57:39 +0000 |
---|---|---|
committer | Keith Davis Arm <keith.davis@arm.com> | 2019-02-26 15:14:23 +0000 |
commit | 07d58c77e174a8e041c1facc61637e85c6cc0bf6 (patch) | |
tree | 7644794f68256e48bd9f1c195bbe9f256def213d /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 762778817f6567f405d5d705f9c2131bab799e66 (diff) | |
download | armnn-07d58c77e174a8e041c1facc61637e85c6cc0bf6.tar.gz |
IVGCVSW-2429 Add Detection PostProcess Parser to TensorFlow Lite Parser
* Added additional custom option use_regular_nms to JSON string
* Renamed use_regular_non_max_suppression to use_regular_nms in parser
Change-Id: I5431eabade6337f6f8c3372670862e32e9cda63c
Signed-off-by: keidav01 <keith.davis@arm.com>
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index bc6316ce7e..cd0e9214c2 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -1771,9 +1771,9 @@ void TfLiteParser::ParseDetectionPostProcess(size_t subgraphIndex, size_t operat desc.m_ScaleX = m["x_scale"].AsFloat(); desc.m_ScaleY = m["y_scale"].AsFloat(); - if (!(m["use_regular_non_max_suppression"].IsNull())) + if (!(m["use_regular_nms"].IsNull())) { - desc.m_UseRegularNms = m["use_regular_non_max_suppression"].AsBool(); + desc.m_UseRegularNms = m["use_regular_nms"].AsBool(); } if (!(m["detections_per_class"].IsNull())) { |