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

2 min read time

Applies to Linux

• include

op_non_max_suppression.hpp

Go to the documentation of this file.

1 

23 #pragma once

24 

25 #include <operator_types.h>

26 

27 #include "core/tensor.hpp"

28 #include "i_operator.hpp"

29 

30 namespace roccv {

35 class NonMaximumSuppression final : public IOperator {

36  public:

41  NonMaximumSuppression() {}

42 

47  ~NonMaximumSuppression() {}

48 

95  void operator()(hipStream_t stream, const Tensor& input, const Tensor& output, const Tensor& scores,

96  float scoreThreshold, float iouThreshold, const eDeviceType device = eDeviceType::GPU) const;

97 };

98 } // namespace roccv

roccv::IOperator

Interface class for all roccv operators.

Definition: i_operator.hpp:31

roccv::NonMaximumSuppression

Class for managing the NMS operator.

Definition: op_non_max_suppression.hpp:35

roccv::NonMaximumSuppression::NonMaximumSuppression

NonMaximumSuppression()

Construct a new Op Non Max Suppression object.

Definition: op_non_max_suppression.hpp:41

roccv::NonMaximumSuppression::operator()

void operator()(hipStream_t stream, const Tensor &input, const Tensor &output, const Tensor &scores, float scoreThreshold, float iouThreshold, const eDeviceType device=eDeviceType::GPU) const

Executes the Non-Maximum Suppression operation. This object performs Non-Maximum Suppression on bound...

roccv::NonMaximumSuppression::~NonMaximumSuppression

~NonMaximumSuppression()

Destroy the Op Non Max Suppression object.

Definition: op_non_max_suppression.hpp:47

roccv::Tensor

Definition: tensor.hpp:37

i_operator.hpp

roccv

Definition: strided_data_wrap.hpp:33

operator_types.h

tensor.hpp

eDeviceType

eDeviceType

Describes the device type. Used to determine where Tensor data should be allocated and whether operat...

Definition: util_enums.h:69

eDeviceType::GPU

@ GPU