昇腾社区首页
中文
注册

RoiExtractor

功能

从特征映射列表中获取RoI(Region of Interest)特征矩阵。

输入

  • features:输入特征Tensor,数据类型:float、float16。shape维度为4维。
  • rois:输入感兴趣区域tensor,数据类型:float、float16。

输出

y:输出Tensor,数据类型:float,float16。shape同输入features一致。

属性

  • finest_scale(可选):数据类型为int。默认值56,最精细尺度。
  • roi_scale_factor(可选):数据类型为float。默认值0,感兴趣区域(ROI)缩放相关的因子。
  • spatial_scale(可选):数据类型为float。空间尺度因子。
  • pooled_height(可选):数据类型为int。默认值7,池化操作中的高度维度。
  • pooled_width(可选):数据类型为int。默认值7,池化操作中的宽度维度。
  • sample_num(可选):数据类型为int。默认值0,采样操作中的样本数量。
  • pool_mode(可选):数据类型为string。池化模式。
  • aligned(可选):数据类型为bool。默认true,是否对齐操作。

约束

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16