本文主要是介绍Faster RCNN 推理 从头写 java (六) NMS (非最大值抑制),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
- 1. 图片预处理
- 2. RPN网络预测
- 3. RPN to ROIs
- 4. Classifier 网络预测
- 5. Classifier网络输出对 ROIs过滤与修正
6. NMS (非最大值抑制)
- 7. 坐标转换为原始图片维度
一: 输入输出
输入:
- candidate_bboxes: shape 为 [M, 4]
- candidate_probs: shape 为 [M, ]
输出:
- bboxes_on_resize: VGG16图片输入维度下的bbox, shape为 [M, 4] N 表示未知
- result_probs: 最终boxes的概率, shape为 [M, ] M 表示未知
这篇关于Faster RCNN 推理 从头写 java (六) NMS (非最大值抑制)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!