distribute_fpn_proposals

paddle.fluid.layers.distribute_fpn_proposals(fpn_rois, min_level, max_level, refer_level, refer_scale, name=None)[source]

This op only takes LoDTensor as input. In Feature Pyramid Networks (FPN) models, it is needed to distribute all proposals into different FPN level, with respect to scale of the proposals, the referring scale and the referring level. Besides, to restore the order of proposals, we return an array which indicates the original index of rois in current proposals. To compute FPN level for each roi, the formula is given as follows:

\[ \begin{align}\begin{aligned}roi\_scale &= \sqrt{BBoxArea(fpn\_roi)}\\level = floor(&\log(\frac{roi\_scale}{refer\_scale}) + refer\_level)\end{aligned}\end{align} \]

where BBoxArea is a function to compute the area of each roi.

Parameters
  • fpn_rois (Variable) – 2-D Tensor with shape [N, 4] and data type is float32 or float64. The input fpn_rois.

  • min_level (int32) – The lowest level of FPN layer where the proposals come from.

  • max_level (int32) – The highest level of FPN layer where the proposals come from.

  • refer_level (int32) – The referring level of FPN layer with specified scale.

  • refer_scale (int32) – The referring scale of FPN layer with specified level.

  • name (str, optional) – For detailed information, please refer to Name. Usually name is no need to set and None by default.

Returns

multi_rois(List) : A list of 2-D LoDTensor with shape [M, 4] and data type of float32 and float64. The length is max_level-min_level+1. The proposals in each FPN level.

restore_ind(Variable): A 2-D Tensor with shape [N, 1], N is the number of total rois. The data type is int32. It is used to restore the order of fpn_rois.

Return type

Tuple

Examples

import paddle.fluid as fluid
fpn_rois = fluid.data(
    name='data', shape=[None, 4], dtype='float32', lod_level=1)
multi_rois, restore_ind = fluid.layers.distribute_fpn_proposals(
    fpn_rois=fpn_rois,
    min_level=2,
    max_level=5,
    refer_level=4,
    refer_scale=224)