技术栈
faster_rcnn
Mr-MegRob
1 年前
深度学习
·
keras
·
faster_rcnn
保姆级 Keras 实现 Faster R-CNN 十四 (预测)
上一篇 文章中我们完成了 Faster R-CNN 训练的功能, 现在到了预测部分了因为在预测的时候并不需标签, 所以 RoiLabelLayer 就不需要了, 也不需要将标签与 rcnn_cls, rcnn_reg 组合. 模型变得更简单了. 以下是用于预测的模型, 相对的是 上一篇 文章中的训练模型
Mr-MegRob
1 年前
深度学习
·
keras
·
faster_rcnn
保姆级 Keras 实现 Faster R-CNN 十
上一篇 文章中我们看到了 RPN 网络的预测效果还是不错的, 但是还是有很多目标是判断错误的, 而且回归修正还有一定的偏差. 这两个问题是正常的, 因为 RPN 的功能就是尽可能地找出可能是目标的区域, 来代替 Selective Search 的功能. 所以也不期望它能把分类和回归的事情都做完了. 接下来就是把 RPN 的输出变成下一阶段 Fast R-CNN 的输入来分类目标和再次调整其位置. 在这之前, 需要实现计算建议区域矩形的功能