box_clip

paddle.fluid.layers. box_clip ( input, im_info, name=None ) [源代码]

将检测框框剪切为 im_info 给出的大小。对于每个输入框，公式如下：

xmin = max(min(xmin, im_w - 1), 0)
ymin = max(min(ymin, im_h - 1), 0)
xmax = max(min(xmax, im_w - 1), 0)
ymax = max(min(ymax, im_h - 1), 0)

其中im_w和im_h是通过im_info计算的：

im_h = round(height / scale)
im_w = round(weight / scale)

参数：

• input (Variable) – 维度为[N_1, N_2, …, N_k, 4]的多维Tensor，其中最后一维为box坐标维度。数据类型为float32或float64。

• im_info (Variable) – 维度为[N, 3]的2-D Tensor，N为输入图片个数。具有（高度height，宽度width，比例scale）图像的信息，其中高度和宽度是输入大小，比例是输入大小和原始大小的比率。数据类型为float32或float64。

• name (str，可选) – 具体用法请参见 Name ，一般无需设置，默认值为None。

返回： 表示剪切后的检测框的Tensor或LoDTensor，数据类型为float32或float64，形状与输入检测框相同

返回类型：Variable

代码示例

import paddle.fluid as fluid
boxes = fluid.data(
    name='boxes', shape=[None, 8, 4], dtype='float32', lod_level=1)
im_info = fluid.data(name='im_info', shape=[None, 3])
out = fluid.layers.box_clip(
    input=boxes, im_info=im_info)