开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

从特征图中获取ROI特征矩阵。自定义Faster R-CNN算子。

[object Object]
[object Object]
[object Object]
  • features ([object Object]):必选参数,表示待处理的特征图,要求4维张量。
  • rois ([object Object]):必选参数,表示ROI位置,支持shape为(N, 4)的2维张量。“N”表示ROI的数量,“4”表示ROI所在图像的index,分别为“x0”、“y0”、“x1”和“y1”。
  • spatial_scale ([object Object]):必选参数,指定[object Object]与原始图像的缩放比率。数据类型支持[object Object]
  • pooled_height ([object Object]):必选参数,指定输出图像的高度。数据类型支持[object Object]
  • pooled_width ([object Object]):必选参数,输出图像的宽度。数据类型支持[object Object]
  • sample_num ([object Object]): 必选参数,默认值为2。指定每个输出元素在H和W方向上的采样频率。若此属性设置为0,则采样频率等于“rois”的向上取整值(一个浮点数)。数据类型支持[object Object]
  • roi_end_mode ([object Object]):必选参数,默认值为1。数据类型支持[object Object]
[object Object]

该算子实现与numpy实现一致,相较于cuda反向存在精度问题。

[object Object]
[object Object]