关于在yolox上增加角度分支的问题

[Axkgag]

trafficstars

您好，我最近也在尝试自己修改yolox网络来实现旋转目标的检测任务，但在角度回归上遇到了些问题，想向您请教一下。 yolox原版的检测头是解耦的，将分类和检测框分作两个分支，我在做角度回归的时候是另外增加一个角度回归分支，output_channels=1，损失函数使用的smooth l1 loss。但角度损失下降的很慢，其他损失基本收敛的情况下，角度损失依旧没有收敛... 想请教一下您，这有没有什么好的办法

[DDGRCF]

请检查一下角度的定义是否有问题

Axkgag @.***> 于2022年11月10日周四 14:13写道：

您好，我最近也在尝试自己修改yolox网络来实现旋转目标的检测任务，但在角度回归上遇到了些问题，想向您请教一下。 yolox原版的检测头是解耦的，将分类和检测框分作两个分支，我在做角度回归的时候是另外增加一个角度回归分支，output_channels=1，损失函数使用的smooth l1 loss。但角度损失下降的很慢，其他损失基本收敛的情况下，角度损失依旧没有收敛... 想请教一下您，这有没有什么好的办法

— Reply to this email directly, view it on GitHub https://github.com/DDGRCF/YOLOX_OBB/issues/32, or unsubscribe https://github.com/notifications/unsubscribe-auth/APFM3AGTNEXS53SHX3RXICLWHSG7LANCNFSM6AAAAAAR4GFUHQ . You are receiving this because you are subscribed to this thread.Message ID: @.***>

[Axkgag]

十分感谢抽空回答。我后来自己找到问题了，是因为yolox自带的数据增强（随机裁剪随机旋转之类的）影响了训练数据的角度标注，关掉之后就能够正常收敛了。