从特征图中获取ROI特征矩阵。自定义Faster R-CNN算子。
[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]。
该算子实现与numpy实现一致,相较于cuda反向存在精度问题。
[object Object]