/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_bnd_box.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_bnd_box.hpp Source File#

2 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_bnd_box.hpp Source File

include

Go to the documentation of this file.

  
 #pragma once
 #include <hip/hip_runtime.h>
 #include <operator_types.h>
  
 #include <i_operator.hpp>
 #include <vector>
  
 #include "core/tensor.hpp"
  
 namespace roccv {
 class BndBox final : public IOperator {
    public:
     BndBox();
  
     ~BndBox();
  
     void operator()(hipStream_t stream, const roccv::Tensor& input, const roccv::Tensor& output,
                     const BndBoxes_t bnd_boxes, eDeviceType device = eDeviceType::GPU);
  
     void generateRects(std::vector<Rect_t>& rects, const BndBoxes_t& bnd_boxes, int64_t height, int64_t width);
 };
 }  // namespace roccv