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... | |
Public Member Functions inherited from roccv::IOperator | |
| 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
Public Member Functions inherited from