Rotated FCOS

Rotated FCOS

由于前段时间看了一下H2Rbox网络,复习一下FCOS
https://arxiv.org/abs/1904.01355
如标题所示,FCOS用于旋转框检测,那么直接上代码(基于mmrotate)
网络大体结构:
img -> backbone -> neck -> head.forward_train
直接看head.forward_train

RotatedFCOSHead(RotatedAnchorFreeHead)

forward函数调用multi_apply(self.forward_single)所以下面关注forward_single函数
首先送入两个分支分别预测:
cls_score分支
bbox_pred分支
保留bbox_pred上一部特征图,直接利用卷积得到centerness,并列的,同时得到angle_pred

这里的一些数据用于loss
之后返回loss即可

如果是测试,需要使用get_bboxes函数,decode之后,multiclass_nms一下即能得到最终框
注意的是,loss计算中需要做一个label assignment问题,不得不说一阶段确实方便,gt中的直接设为正例即可。

猜你喜欢

转载自blog.csdn.net/fei_YuHuo/article/details/127590548