faster rcnn代码解读参考
:https://github.com/adityaarun1/pytorch_fast-er_rcnn
https://github.com/jwyang/faster-rcnn.pytorch
这里的NMS(non maximum suppression)用的是torchvision的nms,直接引用的,没有使用cpp_extension
步骤:
- 根据score排序(得分最大的当然最先被保留下来)。
- 遍历出一个bbox(按照得分从大到小获取),如果该bbox与保留下来的最大bbox的iou阈值小于阈值threshold,我们就保留下来,否则不保留