box_clip

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

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

  1. xmin = max(min(xmin, im_w - 1), 0)
  2. ymin = max(min(ymin, im_h - 1), 0)
  3. xmax = max(min(xmax, im_w - 1), 0)
  4. ymax = max(min(ymax, im_h - 1), 0)

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

  1. im_h = round(height / scale)
  2. 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

代码示例

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