| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
Add RoiAlign operator implementation (issue #27) Co-authored-by: kbryantttt<2305258805@qq.com> # message auto-generated for no-merge-commit merge: !145 merge roialign into master Add RoiAlign operator implementation (issue #27) Created-by: kbryantttt Commit-by: kbryantttt Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 来源：哈尔滨工业大学算子团队 使用AscendC实现了RoiAlign算子，实现了AscendC实现的RoiAlign算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配 ROIAlign 是目标检测/实例分割等视觉任务中的关键算子，用于将不同大小的 ROI（Region of Interest）在特征图上对齐采样并输出固定尺寸特征（如 7×7、14×14），常用于 Faster R-CNN、Mask R-CNN 等结构中。 ROIAlign 相比 ROIPool 的价值在于： 通过双线性插值避免量化误差，提升检测/分割精度； 支持可配置 pooled_height/pooled_width，输出对齐到固定尺寸，便于后续 head 网络处理； 支持 sampling_ratio 控制每个 bin 内采样点数； 支持 float32。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 关联Issue [#27](https://gitcode.com/cann/ops-cv/issues/27) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 采用aclnn调用测试 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-cv!145 | 2 个月前 |