inference-engine/src/inference_engine/builders/ie_simpler_nms_layer.cpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <builders/ie_simpler_nms_layer.hpp>
   6 #include <ie_cnn_layer_builder.h>
   7 #include <vector>
   8 #include <string>
   9
  10 using namespace InferenceEngine;
  11
  12 Builder::SimplerNMSLayer::SimplerNMSLayer(const std::string& name): LayerDecorator("SimplerNMS", name) {
  13     getLayer()->getOutputPorts().resize(1);
  14 }
  15
  16 Builder::SimplerNMSLayer::SimplerNMSLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
  17     checkType("SimplerNMS");
  18 }
  19
  20 Builder::SimplerNMSLayer::SimplerNMSLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
  21     checkType("SimplerNMS");
  22 }
  23
  24 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setName(const std::string& name) {
  25     getLayer()->setName(name);
  26     return *this;
  27 }
  28 const std::vector<Port>& Builder::SimplerNMSLayer::getInputPorts() const {
  29     return getLayer()->getInputPorts();
  30 }
  31 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setInputPorts(const std::vector<Port>& ports) {
  32     getLayer()->getInputPorts() = ports;
  33     return *this;
  34 }
  35 const Port& Builder::SimplerNMSLayer::getOutputPort() const {
  36     return getLayer()->getOutputPorts()[0];
  37 }
  38 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setOutputPort(const Port& port) {
  39     getLayer()->getOutputPorts()[0] = port;
  40     return *this;
  41 }
  42
  43 size_t Builder::SimplerNMSLayer::getPreNMSTopN() const {
  44     return getLayer()->getParameters().at("pre_nms_topn");
  45 }
  46 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setPreNMSTopN(size_t topN) {
  47     getLayer()->getParameters()["pre_nms_topn"] = topN;
  48     return *this;
  49 }
  50 size_t Builder::SimplerNMSLayer::getPostNMSTopN() const {
  51     return getLayer()->getParameters().at("post_nms_topn");
  52 }
  53 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setPostNMSTopN(size_t topN) {
  54     getLayer()->getParameters()["post_nms_topn"] = topN;
  55     return *this;
  56 }
  57 size_t Builder::SimplerNMSLayer::getFeatStride() const {
  58     return getLayer()->getParameters().at("feat_stride");
  59 }
  60 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setFeatStride(size_t featStride) {
  61     getLayer()->getParameters()["feat_stride"] = featStride;
  62     return *this;
  63 }
  64 size_t Builder::SimplerNMSLayer::getMinBoxSize() const {
  65     return getLayer()->getParameters().at("min_bbox_size");
  66 }
  67 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setMinBoxSize(size_t minSize) {
  68     getLayer()->getParameters()["min_bbox_size"] = minSize;
  69     return *this;
  70 }
  71 size_t Builder::SimplerNMSLayer::getScale() const {
  72     return getLayer()->getParameters().at("scale");
  73 }
  74 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setScale(size_t scale) {
  75     getLayer()->getParameters()["scale"] = scale;
  76     return *this;
  77 }
  78
  79 float Builder::SimplerNMSLayer::getCLSThreshold() const {
  80     return getLayer()->getParameters().at("cls_threshold");
  81 }
  82 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setCLSThreshold(float threshold) {
  83     getLayer()->getParameters()["cls_threshold"] = threshold;
  84     return *this;
  85 }
  86 float Builder::SimplerNMSLayer::getIOUThreshold() const {
  87     return getLayer()->getParameters().at("iou_threshold");
  88 }
  89 Builder::SimplerNMSLayer& Builder::SimplerNMSLayer::setIOUThreshold(float threshold) {
  90     getLayer()->getParameters()["iou_threshold"] = threshold;
  91     return *this;
  92 }
  93
  94 REG_CONVERTER_FOR(SimplerNMS, [](const CNNLayerPtr& cnnLayer, Builder::Layer& layer) {
  95     layer.getParameters()["iou_threshold"] = cnnLayer->GetParamAsFloat("iou_threshold");
  96     layer.getParameters()["cls_threshold"] = cnnLayer->GetParamAsFloat("cls_threshold");
  97     layer.getParameters()["scale"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("scale"));
  98     layer.getParameters()["min_bbox_size"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("min_bbox_size"));
  99     layer.getParameters()["feat_stride"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("feat_stride"));
 100     layer.getParameters()["pre_nms_topn"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("pre_nms_topn"));
 101     layer.getParameters()["post_nms_topn"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("post_nms_topn"));
 102 });