diff options
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())) { |