BndBox Class Reference#
2 min read time
Applies to Linux
roccv::BndBox Class Referencefinal
Class for managing the BndBox Operator. More...
#include <op_bnd_box.hpp>
Inheritance diagram for roccv::BndBox:

Public Member Functions | |
BndBox () | |
Constructs a BndBox object. More... | |
void | generateRects (std::vector< Rect_t > &rects, const BndBoxes_t &bnd_boxes, int64_t height, int64_t width) |
void | operator() (hipStream_t stream, const roccv::Tensor &input, const roccv::Tensor &output, const BndBoxes_t bnd_boxes, eDeviceType device=eDeviceType::GPU) |
Executes the BndBox operation. BndBox can be used to draw bounding boxes on images in a tensor. More... | |
~BndBox () | |
Destroy the BndBox object. More... | |
![]() | |
virtual | ~IOperator () |
Detailed Description
Class for managing the BndBox Operator.
Constructor & Destructor Documentation
◆ BndBox()
roccv::BndBox::BndBox | ( | ) |
Constructs a BndBox object.
◆ ~BndBox()
roccv::BndBox::~BndBox | ( | ) |
Destroy the BndBox object.
Member Function Documentation
◆ generateRects()
void roccv::BndBox::generateRects | ( | std::vector< Rect_t > & | rects, |
const BndBoxes_t & | bnd_boxes, | ||
int64_t | height, | ||
int64_t | width | ||
) |
◆ operator()()
void roccv::BndBox::operator() | ( | hipStream_t | stream, |
const roccv::Tensor & | input, | ||
const roccv::Tensor & | output, | ||
const BndBoxes_t | bnd_boxes, | ||
eDeviceType | device = eDeviceType::GPU |
||
) |
Executes the BndBox operation. BndBox can be used to draw bounding boxes on images in a tensor.
Limitations:
Input: Supported TensorLayout(s): [NHWC, HWC] Channels: [3, 4] Supported DataType(s): [U8]
Output: Supported TensorLayout(s): [NHWC, HWC] Channels: [3, 4] Supported DataType(s): [U8]
Input/Output dependency:
Property | Input == Output -------------- | ------------- TensorLayout | Yes DataType | Yes Channels | Yes Width | Yes Height | Yes Batch | Yes
- Parameters
-
[in] stream The HIP stream to run this operation on. [in] input Input tensor. [out] output Output tensor. [in] bnd_boxes Bounding boxes to apply to input tensor. [in] device The device which this operation should run on. (Default: eDeviceType::GPU)
The documentation for this class was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/op_bnd_box.hpp