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

2 min read time

Applies to Linux

• include

op_center_crop.hpp

Go to the documentation of this file.

1 

22 #pragma once

23 

24 #include <i_operator.hpp>

25 

26 #include "core/tensor.hpp"

27 #include "operator_types.h"

28 #include "op_custom_crop.hpp"

29 

30 

31 namespace roccv {

36 class CenterCrop final : public IOperator {

37  public:

42  CenterCrop() {}

43 

48  ~CenterCrop() {}

49 

85  void operator()(hipStream_t stream, const Tensor &input, const Tensor &output, const Size2D cropSize,

86  const eDeviceType device = eDeviceType::GPU) const;

87 

88  private:

89  CustomCrop m_op;

90 };

91 } // namespace roccv

roccv::CenterCrop

Class for managing the Center Crop operator.

Definition: op_center_crop.hpp:36

roccv::CenterCrop::~CenterCrop

~CenterCrop()

Destroy the Op CenterCrop object.

Definition: op_center_crop.hpp:48

roccv::CenterCrop::operator()

void operator()(hipStream_t stream, const Tensor &input, const Tensor &output, const Size2D cropSize, const eDeviceType device=eDeviceType::GPU) const

Construct a new Op Center Crop object. The object can be used to crop a region of interest from the c...

roccv::CenterCrop::CenterCrop

CenterCrop()

Construct a new Op CenterCrop object.

Definition: op_center_crop.hpp:42

roccv::CustomCrop

Class for managing the Custom Crop operator.

Definition: op_custom_crop.hpp:35

roccv::IOperator

Interface class for all roccv operators.

Definition: i_operator.hpp:31

roccv::Tensor

Definition: tensor.hpp:37

i_operator.hpp

roccv

Definition: strided_data_wrap.hpp:33

op_custom_crop.hpp

operator_types.h

roccv::Size2D

Describes the 2D dimensions of an image.

Definition: operator_types.h:130

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