#include <nms_cuda.h>

Collaboration diagram for apollo::perception::lidar::NmsCuda:

Public Member Functions
	NmsCuda (const int num_threads, const int num_box_corners, const float nms_overlap_threshold)
	Constructor. More...

void	DoNmsCuda (const int host_filter_count, float dev_sorted_box_for_nms, int out_keep_inds, int *out_num_to_keep)
	GPU Non-Maximum Suppresion for network output. More...

Constructor & Destructor Documentation

◆ NmsCuda()

apollo::perception::lidar::NmsCuda::NmsCuda	(	const int	num_threads,
		const int	num_box_corners,
		const float	nms_overlap_threshold
	)

Constructor.

Parameters

[in]	num_threads	Number of threads when launching cuda kernel
[in]	num_box_corners	Number of corners for 2D box
[in]	nms_overlap_threshold	IOU threshold for NMS

Captital variables never change after the compile, Non-captital variables could be chaned through rosparam

void apollo::perception::lidar::NmsCuda::DoNmsCuda	(	const int	host_filter_count,
		float *	dev_sorted_box_for_nms,
		int *	out_keep_inds,
		int *	out_num_to_keep
	)

GPU Non-Maximum Suppresion for network output.

Parameters

[in]	host_filter_count	Number of filtered output
[in]	dev_sorted_box_for_nms	Bounding box output sorted by score
[out]	out_keep_inds	Indexes of selected bounding box
[out]	out_num_to_keep	Number of kept bounding boxes

NMS in GPU and postprocessing for selecting box in CPU

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