apollo::perception::inference::ROIPoolingLayer< Dtype > Class Template Reference

#include <roipooling_layer.h>

Inheritance diagram for apollo::perception::inference::ROIPoolingLayer< Dtype >:

Collaboration diagram for apollo::perception::inference::ROIPoolingLayer< Dtype >:

Public Member Functions
	ROIPoolingLayer (int pooled_h, int pooled_w, bool use_floor, float spatial_scale, int channels, int max_objs=1000)
void	ForwardGPU (const std::vector< std::shared_ptr< base::Blob< Dtype >>> &bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &top)
void	ForwardCPU (const std::vector< std::shared_ptr< base::Blob< Dtype >>> &bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &top)
Public Member Functions inherited from apollo::perception::inference::Layer< Dtype >
	Layer ()=default
virtual	~Layer ()=default
void	set_gpu_id (const int &gpu_id)

Additional Inherited Members
Protected Attributes inherited from apollo::perception::inference::Layer< Dtype >
int	gpu_id_ = 0

Constructor & Destructor Documentation

ROIPoolingLayer()

template<typename Dtype >

apollo::perception::inference::ROIPoolingLayer< Dtype >::ROIPoolingLayer ( int  pooled_h, int  pooled_w, bool  use_floor, float  spatial_scale, int  channels, int  max_objs = 1000  )

inline

Member Function Documentation

ForwardCPU()

template<typename Dtype >

void apollo::perception::inference::ROIPoolingLayer< Dtype >::ForwardCPU ( const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  top  )

virtual

Implements apollo::perception::inference::Layer< Dtype >.

ForwardGPU()

template<typename Dtype >

void apollo::perception::inference::ROIPoolingLayer< Dtype >::ForwardGPU ( const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  bottom, const std::vector< std::shared_ptr< base::Blob< Dtype >>> &  top  )

virtual

Implements apollo::perception::inference::Layer< Dtype >.

---

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

• modules/perception/inference/operators/roipooling_layer.h