FasterRCNN源码解析——数据在RPN的传输路线数据在RPN的传输路线
文章目录数据在RPN的传输路线我们经过RPNHead后我们经过anchor生成器之后求得一个batch中有多少图片计算每个预测特征层上对应的anchors数量调整内部tensor格式以及shape将预测的bbox regression参数应用到anchors上得到最终预测bbox坐标筛除小boxes框,nms处理,根据预测概率获取前post_nms_top_n个目标数据在RPN的传输路线即RegionProposalNetwork类中的forward函数通过在开头所传入的三个数据images,f