modules/dnn/src/layers/const_layer.cpp

   1 // This file is part of OpenCV project.
   2 // It is subject to the license terms in the LICENSE file found in the top-level directory
   3 // of this distribution and at http://opencv.org/license.html.
   4
   5 // Copyright (C) 2018, Intel Corporation, all rights reserved.
   6 // Third party copyrights are property of their respective owners.
   7
   8 #include "../precomp.hpp"
   9 #include "../op_inf_engine.hpp"
  10 #include "layers_common.hpp"
  11
  12 #ifdef HAVE_OPENCL
  13 #include "opencl_kernels_dnn.hpp"
  14 #endif
  15
  16 namespace cv { namespace dnn {
  17
  18 class ConstLayerImpl CV_FINAL : public ConstLayer
  19 {
  20 public:
  21     ConstLayerImpl(const LayerParams& params)
  22     {
  23         setParamsFrom(params);
  24         CV_Assert(blobs.size() == 1);
  25     }
  26
  27     virtual bool supportBackend(int backendId) CV_OVERRIDE
  28     {
  29         return backendId == DNN_BACKEND_OPENCV || backendId == DNN_BACKEND_INFERENCE_ENGINE;
  30     }
  31
  32     virtual bool getMemoryShapes(const std::vector<MatShape> &inputs,
  33                                  const int requiredOutputs,
  34                                  std::vector<MatShape> &outputs,
  35                                  std::vector<MatShape> &internals) const CV_OVERRIDE
  36     {
  37         CV_Assert(inputs.empty());
  38         outputs.assign(1, shape(blobs[0]));
  39         return false;
  40     }
  41
  42 #ifdef HAVE_OPENCL
  43     bool forward_ocl(InputArrayOfArrays inps, OutputArrayOfArrays outs, OutputArrayOfArrays internals)
  44     {
  45         std::vector<UMat> outputs;
  46         outs.getUMatVector(outputs);
  47         if (outs.depth() == CV_16S)
  48             convertFp16(blobs[0], outputs[0]);
  49         else
  50             blobs[0].copyTo(outputs[0]);
  51         return true;
  52     }
  53 #endif
  54
  55     void forward(InputArrayOfArrays inputs_arr, OutputArrayOfArrays outputs_arr, OutputArrayOfArrays internals_arr) CV_OVERRIDE
  56     {
  57         CV_TRACE_FUNCTION();
  58         CV_TRACE_ARG_VALUE(name, "name", name.c_str());
  59
  60         CV_OCL_RUN(IS_DNN_OPENCL_TARGET(preferableTarget),
  61                    forward_ocl(inputs_arr, outputs_arr, internals_arr))
  62
  63         std::vector<Mat> outputs;
  64         outputs_arr.getMatVector(outputs);
  65         blobs[0].copyTo(outputs[0]);
  66     }
  67
  68 #ifdef HAVE_INF_ENGINE
  69     virtual Ptr<BackendNode> initInfEngine(const std::vector<Ptr<BackendWrapper> >&) CV_OVERRIDE
  70     {
  71         InferenceEngine::Builder::ConstLayer ieLayer(name);
  72         ieLayer.setData(wrapToInfEngineBlob(blobs[0]));
  73         return Ptr<BackendNode>(new InfEngineBackendNode(ieLayer));
  74     }
  75 #endif  // HAVE_INF_ENGINE
  76 };
  77
  78 Ptr<Layer> ConstLayer::create(const LayerParams& params)
  79 {
  80     return Ptr<Layer>(new ConstLayerImpl(params));
  81 }
  82
  83 }}  // namespace cv::dnn