Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RTCdet的介绍有吗 #4

Open
jie311 opened this issue Sep 7, 2023 · 1 comment
Open

RTCdet的介绍有吗 #4

jie311 opened this issue Sep 7, 2023 · 1 comment

Comments

@jie311
Copy link

jie311 commented Sep 7, 2023

No description provided.

@yjh0410
Copy link
Owner

yjh0410 commented Sep 7, 2023

@jie311 暂时没有。不过,我在这里大概介绍一下吧。

网络结构

RTCDet的网络结构近似于我在这个项目里实现的YOLOv7的结构,核心模块是ELANBlock,可以直接看代码,不难理解;Neck用的就是SPP模块,PaFPN用的是YOLOv7风格的PaFPN,但把通道数适当扩大了2倍,并且针对PaFPN重新设计了一个ELANBlockFPN,整个PaFPN结构要比v7的厚一些,最后就是YOLOX风格的解耦检测头。
最后,设计了width和depth因子,来设计不同规模的RTCDet检测器;

训练配置

这一块适当参考了MMYOLO的RTMDet,使用AdamW优化器,weight decay=0.5,baselr=0.001(对应的batch size为64),根据batch size的大小自动调整学习率;使用了Model EMA;Cosine学习衰减策略;数据增强使用的是YOLOX风格的:

基本就是这么个情况。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants