Apollo  6.0
Open source self driving car software
Public Member Functions | List of all members
apollo::perception::lidar::AnchorMaskCuda Class Reference

#include <anchor_mask_cuda.h>

Collaboration diagram for apollo::perception::lidar::AnchorMaskCuda:
Collaboration graph

Public Member Functions

 AnchorMaskCuda (const int num_threads, const int num_inds_for_scan, const int num_anchor, const float min_x_range, const float min_y_range, const float pillar_x_size, const float pillar_y_size, const int grid_x_size, const int grid_y_size)
 Constructor. More...
 
void DoAnchorMaskCuda (int *dev_sparse_pillar_map, int *dev_cumsum_along_x, int *dev_cumsum_along_y, const float *dev_box_anchors_min_x, const float *dev_box_anchors_min_y, const float *dev_box_anchors_max_x, const float *dev_box_anchors_max_y, int *dev_anchor_mask)
 call cuda code for making anchor mask More...
 

Constructor & Destructor Documentation

◆ AnchorMaskCuda()

apollo::perception::lidar::AnchorMaskCuda::AnchorMaskCuda ( const int  num_threads,
const int  num_inds_for_scan,
const int  num_anchor,
const float  min_x_range,
const float  min_y_range,
const float  pillar_x_size,
const float  pillar_y_size,
const int  grid_x_size,
const int  grid_y_size 
)

Constructor.

Parameters
[in]num_threadsNumber of threads per block
[in]num_inds_for_scanNumber of indexes for scan(cumsum)
[in]num_anchorNumber of anchors in total
[in]min_x_rangeMinimum x value for point cloud
[in]min_y_rangeMinimum y value for point cloud
[in]pillar_x_sizeSize of x-dimension for a pillar
[in]pillar_y_sizeSize of y-dimension for a pillar
[in]grid_x_sizeNumber of pillars in x-coordinate
[in]grid_y_sizeNumber of pillars in y-coordinate

Captital variables never change after the compile

Member Function Documentation

◆ DoAnchorMaskCuda()

void apollo::perception::lidar::AnchorMaskCuda::DoAnchorMaskCuda ( int *  dev_sparse_pillar_map,
int *  dev_cumsum_along_x,
int *  dev_cumsum_along_y,
const float *  dev_box_anchors_min_x,
const float *  dev_box_anchors_min_y,
const float *  dev_box_anchors_max_x,
const float *  dev_box_anchors_max_y,
int *  dev_anchor_mask 
)

call cuda code for making anchor mask

Parameters
[in]dev_sparse_pillar_mapGrid map representation for pillar occupancy
[in]dev_cumsum_along_xArray for storing cumsum-ed dev_sparse_pillar_map values
[in]dev_cumsum_along_yArray for storing cumsum-ed dev_cumsum_along_y values
[in]dev_box_anchors_min_xArray for storing min x value for each anchor
[in]dev_box_anchors_min_yArray for storing min y value for each anchor
[in]dev_box_anchors_max_xArray for storing max x value for each anchor
[in]dev_box_anchors_max_yArray for storing max y value for each anchor
[in]dev_box_anchors_max_yArray for storing max y value for each anchor
[out]dev_anchor_maskAnchor mask for filtering the network output

dev_* means device memory. Make a mask for filtering pillar occupancy area


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