Blame view

3rdparty/opencv-4.5.4/modules/gapi/src/backends/cpu/gnnparsers.hpp 1.09 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
  // This file is part of OpenCV project.
  // It is subject to the license terms in the LICENSE file found in the top-level directory
  // of this distribution and at http://opencv.org/license.html.
  //
  // Copyright (C) 2020 Intel Corporation
  
  #include <opencv2/gapi/infer/parsers.hpp>
  
  #ifndef OPENCV_NNPARSERS_OCV_HPP
  #define OPENCV_NNPARSERS_OCV_HPP
  
  namespace cv
  {
  void ParseSSD(const cv::Mat&  in_ssd_result,
                const cv::Size& in_size,
                const float     confidence_threshold,
                const int       filter_label,
                const bool      alignment_to_square,
                const bool      filter_out_of_bounds,
                std::vector<cv::Rect>& out_boxes,
                std::vector<int>&      out_labels);
  
  void parseYolo(const cv::Mat&  in_yolo_result,
                 const cv::Size& in_size,
                 const float     confidence_threshold,
                 const float     nms_threshold,
                 const std::vector<float>& anchors,
                 std::vector<cv::Rect>& out_boxes,
                 std::vector<int>&      out_labels);
  }
  #endif // OPENCV_NNPARSERS_OCV_HPP