modules/dnn/src/cuda4dnn/kernels/prior_box.hpp

   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 #ifndef OPENCV_DNN_SRC_CUDA4DNN_KERNELS_PRIOR_BOX_HPP
   6 #define OPENCV_DNN_SRC_CUDA4DNN_KERNELS_PRIOR_BOX_HPP
   7
   8 #include "../csl/stream.hpp"
   9 #include "../csl/span.hpp"
  10
  11 #include <cstddef>
  12
  13 namespace cv { namespace dnn { namespace cuda4dnn { namespace kernels {
  14
  15     template <class T>
  16     void generate_prior_boxes(
  17         const csl::Stream& stream,
  18         csl::Span<T> output,
  19         csl::View<float> boxWidth, csl::View<float> boxHeight, csl::View<float> offsetX, csl::View<float> offsetY, float stepX, float stepY,
  20         std::vector<float> variance,
  21         std::size_t numPriors,
  22         std::size_t layerWidth, std::size_t layerHeight,
  23         std::size_t imageWidth, std::size_t imageHeight,
  24         bool normalize, bool clip);
  25
  26 }}}} /* namespace cv::dnn::cuda4dnn::kernels */
  27
  28 #endif /* OPENCV_DNN_SRC_CUDA4DNN_KERNELS_PRIOR_BOX_HPP */