bmcv_nms

该接口用于消除网络计算得到过多的物体框,并找到最佳物体框。

接口形式:

bm_status_t bmcv_nms(bm_handle_t handle,
                     bm_device_mem_t input_proposal_addr,
                     int proposal_size,
                     float nms_threshold,
                     bm_device_mem_t output_proposal_addr)

参数说明:

  • bm_handle_t handle

    输入参数。bm_handle 句柄。

  • bm_device_mem_t input_proposal_addr

    输入参数。输入物体框数据所在地址,输入物体框数据结构为 face_rect_t,详见下面数据结构说明。需要调用 bm_mem_from_system()将数据地址转化成转化为 bm_device_mem_t 所对应的结构。

  • int proposal_size

    输入参数。物体框个数。

  • float nms_threshold

    输入参数。过滤物体框的阈值,分数小于该阈值的物体框将会被过滤掉。

  • bm_device_mem_t output_proposal_addr

    输出参数。输出物体框数据所在地址,输出物体框数据结构为 nms_proposal_t,详见下面数据结构说明。需要调用 bm_mem_from_system() 将数据地址转化成转化为 bm_device_mem_t 所对应的结构。

返回值:

  • BM_SUCCESS: 成功

  • 其他: 失败

数据类型说明:

face_rect_t 描述了一个物体框坐标位置以及对应的分数。

typedef struct
{
    float x1;
    float y1;
    float x2;
    float y2;
    float score;
}face_rect_t;
  • x1 描述了物体框左边缘的横坐标

  • y1 描述了物体框上边缘的纵坐标

  • x2 描述了物体框右边缘的横坐标

  • y2 描述了物体框下边缘的纵坐标

  • score 描述了物体框对应的分数

nms_proposal_t 描述了输出物体框的信息。

  typedef struct
  {
      face_rect_t face_rect[MAX_PROPOSAL_NUM];
      int size;
      int capacity;
      face_rect_t *begin;
      face_rect_t *end;
  } nms_proposal_t;

* face_rect 描述了经过过滤后的物体框信息

* size 描述了过滤后得到的物体框个数

* capacity 描述了过滤后物体框最大个数

* begin 暂不使用

* end 暂不使用

代码示例:

face_rect_t *proposal_rand = new face_rect_t[MAX_PROPOSAL_NUM];
nms_proposal_t *output_proposal = new nms_proposal_t;
int proposal_size =32;
float nms_threshold = 0.2;
for (int i = 0; i < proposal_size; i++)
{
    proposal_rand[i].x1 = 200;
    proposal_rand[i].x2 = 210 ;
    proposal_rand[i].y1 = 200;
    proposal_rand[i].y2 = 210;
    proposal_rand[i].score = 0.23;
}
bmcv_nms(handle,
         bm_mem_from_system(proposal_rand),
         proposal_size,
         nms_threshold,
         bm_mem_from_system(output_proposal));
delete[] proposal_rand;
delete output_proposal;

注意事项:

该 api 可输入的最大 proposal 数为 56000。