[object Object]
按分数降序选择旋转标注框的子集。
[object Object]
- dets(
[object Object]):shape为[num_boxes, 5]的2D浮点张量。 - scores(
[object Object]):shape为[num_boxes]的1D浮点张量,表示每个框对应的一个分数。 - iou_threshold(
[object Object]):表示框与IoU重叠上限阈值的标量。 - scores_threshold(
[object Object]):表示决定何时删除框的分数阈值的标量,默认值为0。 - max_output_size(
[object Object]):标量整数张量,表示非最大抑制下要选择的最大框数。默认值为-1,即不施加任何约束。 - mode(
[object Object]):指定dets布局类型,默认值为0。如果mode设置为0,则dets的输入值为x、y、w、h和角度。如果mode设置为1,则dets的输入值为x1、y1、x2、y2和角度。
- selected_index(
[object Object]):shape为[M]的1D整数张量,表示从[object Object]张量中选定的index,其中M <= max_output_size。 - selected_num(
[object Object]):0D整数张量,表示[object Object]中有效元素的数量。
目前不支持mode=1的场景。
[object Object]