BndBox Class Reference

BndBox Class Reference#

2 min read time

Applies to Linux

rocCV: roccv::BndBox Class Reference
roccv::BndBox Class Referencefinal

Class for managing the BndBox Operator. More...

#include <op_bnd_box.hpp>

Inheritance diagram for roccv::BndBox:
roccv::IOperator

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]streamThe HIP stream to run this operation on.
[in]inputInput tensor.
[out]outputOutput tensor.
[in]bnd_boxesBounding boxes to apply to input tensor.
[in]deviceThe 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