wulele2 / RotateCenterNet

CenterNet基础上实现旋转目标检测

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RotateCenterNet

CenterNet基础上实现旋转目标检测

1.训练

本仓库将所有的遥感数据集:HRSC,SSDD++首先借助DatasetTransform文件夹内的脚本统一将标签转成了[cx,cy,w,h,theta],theta属于[0,180)的形式,即长边表示法。 然后通过更改cfg的配置文件信息,便可进行训练。

2.测试

另外,本人添加了可视化的脚本以及计算map的脚本,以及具有nms的centernet的评估脚本 好久没用了,有点儿乱。见谅。

3.实验结果

最终能在hrsc数据集上借助L1损失达到了87的map;另外,添加了GWD_LOSS,只是训练不稳定,经常nan,希望大佬提PR,解决这个问题

About

CenterNet基础上实现旋转目标检测


Languages

Language:Python 100.0%