bubbliiiing/retinanet-pytorch

有关加入cascade的问题

speechle55 opened this issue · 4 comments

您好,我在训练自己的数据集时,发现了生成box与gt存在一点差距的问题,boss让我加入cascade来解决。但是我看了下cascade的代码,感觉脑子好混乱,里面还涉及了roipooling、RoIAlign等等。请问如果我要在这个retina中加入cascade的话,要在哪里改,大概的思路是什么,谢谢!

……双阶段和单阶段差太大了吧,你不如直接去用cascade的代码,而且,为什么用了cascade就没有差距了呢

因为我们观察了假阳数据后,发现假阳数据并不是真正的假阳,而是bbox与gt的差距有一些大,造成的iou较低,导致被判定成了假阳,我老板说可以用cascade来生成高质量的bbox,这样效果会变好。我目前查到了一些文章,也是在Retina的基础上加入了cascade的**,他们的实验结果也是会对模型有提升,但是他们都没有开源代码,大家也说实现起来很简单。我把链接粘上来您看看。
ConRetinaNet:https://zhuanlan.zhihu.com/p/55416312 原文:https://arxiv.org/pdf/1901.06563.pdf
Cascade RetinaNet:https://blog.csdn.net/qq_41375609/article/details/105193028 原文:https://arxiv.org/abs/1907.06881v1

老板说在代码中接一个新的头就可以,但是我还是没太能写出来。。。