Compute Library
 19.08
CLBoundingBoxTransform Class Reference

Basic function to run CLBoundingBoxTransformKernel. More...

#include <CLBoundingBoxTransform.h>

Collaboration diagram for CLBoundingBoxTransform:
[legend]

Public Member Functions

void configure (const ICLTensor *boxes, ICLTensor *pred_boxes, const ICLTensor *deltas, const BoundingBoxTransformInfo &info)
 Set the input and output tensors. More...
 
- Public Member Functions inherited from ICLSimpleFunction
 ICLSimpleFunction ()
 Default constructor. More...
 
void run () override final
 Run the kernels contained in the function. More...
 
- Public Member Functions inherited from IFunction
virtual ~IFunction ()=default
 Destructor. More...
 
virtual void prepare ()
 Prepare the function for executing. More...
 

Static Public Member Functions

static Status validate (const ITensorInfo *boxes, const ITensorInfo *pred_boxes, const ITensorInfo *deltas, const BoundingBoxTransformInfo &info)
 Static function to check if given info will lead to a valid configuration of CLBoundingBoxTransform. More...
 

Detailed Description

Basic function to run CLBoundingBoxTransformKernel.

This function calls the following OpenCL kernels:

  1. CLBoundingBoxTransformKernel

Definition at line 39 of file CLBoundingBoxTransform.h.

Member Function Documentation

◆ configure()

void configure ( const ICLTensor boxes,
ICLTensor pred_boxes,
const ICLTensor deltas,
const BoundingBoxTransformInfo info 
)

Set the input and output tensors.

Parameters
[in]boxesSource tensor. Bounding box proposals in pixel coordinates. Size(M, 4), format [x1, y1, x2, y2]. Data types supported: F16/F32.
[out]pred_boxesDestination tensor. Pixel coordinates of the transformed bounding boxes. Size (M, 4*K), format [x1, y1, x2, y2]. Data types supported: Same as input
[in]deltasBounding box translations and scales. Size (M, 4*K), format [dx, dy, dw, dh], K is the number of classes. Data types supported: Same as input
[in]infoContains BoundingBox operation information described in BoundingBoxTransformInfo.
Note
Only single image prediction is supported. Height and Width (and scale) of the image will be contained in the BoundingBoxTransformInfo struct.

Definition at line 31 of file CLBoundingBoxTransform.cpp.

32 {
33  // Configure Bounding Box kernel
34  auto k = arm_compute::support::cpp14::make_unique<CLBoundingBoxTransformKernel>();
35  k->configure(boxes, pred_boxes, deltas, info);
36  _kernel = std::move(k);
37 }

References arm_compute::test::validation::info.

◆ validate()

Status validate ( const ITensorInfo boxes,
const ITensorInfo pred_boxes,
const ITensorInfo deltas,
const BoundingBoxTransformInfo info 
)
static

Static function to check if given info will lead to a valid configuration of CLBoundingBoxTransform.

Parameters
[in]boxesSource tensor info. Bounding box proposals in pixel coordinates. Size(M, 4), format [x1, y1, x2, y2]. Data types supported: F16/F32.
[in]pred_boxesDestination tensor info. Pixel coordinates of the transformed bounding boxes. Size (M, 4*K), format [x1, y1, x2, y2]. Data types supported: Same as input
[in]deltasBounding box translations and scales. Size (M, 4*K), format [dx, dy, dw, dh], K is the number of classes. Data types supported: Same as input
[in]infoContains BoundingBox operation information described in BoundingBoxTransformInfo.
Note
Only single image prediction is supported. Height and Width (and scale) of the image will be contained in the BoundingBoxTransformInfo struct.
Returns
a Status

Definition at line 39 of file CLBoundingBoxTransform.cpp.

40 {
41  return CLBoundingBoxTransformKernel::validate(boxes, pred_boxes, deltas, info);
42 }
static Status validate(const ITensorInfo *boxes, const ITensorInfo *pred_boxes, const ITensorInfo *deltas, const BoundingBoxTransformInfo &info)
Static function to check if given info will lead to a valid configuration of CLBoundingBoxTransform.

References arm_compute::test::validation::info, and CLBoundingBoxTransformKernel::validate().


The documentation for this class was generated from the following files: