最近,看了OHEM这篇论文,也是RBG的一篇论文,其是在Fast RCNN上进行改进,目标候选框提取仍然采用Selective Search算法。我打算在Faster RCNN上修改,充分利用RPN网络提取目标候选框。整个修改还比比较方便的。Fast RCNN + OHEM的相关代码论文作者已经在github上开源,地址在:https://github.com/abhi2610/ohem,我们只需在其基础上改进即可。我们知道Faster RCNN有两种常用的训练方式,一种为end-to-end,一种为4步交替训练方式。我们在文中分别给出其prototxt文件的实现。
############################ 待更新, 2017/3/20
Faster-RCNN + OHEM + vgg16 + prototxt文件及相关代码修改
最新推荐文章于 2024-09-25 09:42:07 发布