yjh0410 / RT-ODLab

YOLO Tutorial
MIT License
136 stars 29 forks source link

RTCdet的介绍有吗 #4

Open jie311 opened 11 months ago

yjh0410 commented 11 months ago

@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风格的:

基本就是这么个情况。