最大区域池化

MaxRoiPool - 22

版本

  • 名称: MaxRoiPool (GitHub)

  • 域名: main

  • since_version: 22

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本22起可用。

摘要

ROI最大池化消耗一个输入张量X和感兴趣区域（RoIs），在每个RoI上应用最大池化，以生成形状为(num_rois, channels, pooled_shape[0], pooled_shape[1])的输出4-D张量。

属性

  • pooled_shape - INTS（必填）：

    ROI池化输出形状（高度，宽度）。

  • spatial_scale - FLOAT (默认值为 '1.0'):

    用于将ROI坐标从输入比例转换为池化时使用的比例的空间比例因子。

输入

  • X (异构) - T:

    来自前一个操作符的输入数据张量；图像情况下的维度为（N x C x H x W），其中N是批量大小，C是通道数，H和W是数据的高度和宽度。

  • rois (异构) - T:

    RoIs（感兴趣区域）用于池化。应该是一个形状为 (num_rois, 5) 的二维张量，格式为 [[batch_id, x1, y1, x2, y2], …]。

输出

  • Y (异构) - T:

    RoI 池化输出的 4-D 张量，形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

类型约束

  • T 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ):

    将输入和输出类型限制为浮点张量。

MaxRoiPool - 1

版本

  • 名称: MaxRoiPool (GitHub)

  • 域名: main

  • since_version: 1

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本1起可用。

摘要

ROI最大池化消耗一个输入张量X和感兴趣区域（RoIs），在每个RoI上应用最大池化，以生成形状为(num_rois, channels, pooled_shape[0], pooled_shape[1])的输出4-D张量。

属性

  • pooled_shape - INTS（必填）：

    ROI池化输出形状（高度，宽度）。

  • spatial_scale - FLOAT (默认值为 '1.0'):

    用于将ROI坐标从输入比例转换为池化时使用的比例的空间比例因子。

输入

  • X (异构) - T:

    来自前一个操作符的输入数据张量；图像情况下的维度为（N x C x H x W），其中N是批量大小，C是通道数，H和W是数据的高度和宽度。

  • rois (异构) - T:

    RoIs（感兴趣区域）用于池化。应该是一个形状为 (num_rois, 5) 的二维张量，格式为 [[batch_id, x1, y1, x2, y2], …]。

输出

  • Y (异构) - T:

    RoI 池化输出的 4-D 张量，形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

类型约束

  • T 在 ( tensor(double), tensor(float), tensor(float16) ) 中:

    将输入和输出类型限制为浮点张量。