失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > 《南溪的目标检测学习笔记》——主干网络backbone设计的学习笔记

《南溪的目标检测学习笔记》——主干网络backbone设计的学习笔记

时间:2023-05-29 01:50:31

相关推荐

《南溪的目标检测学习笔记》——主干网络backbone设计的学习笔记

1 南溪推荐的主干网络——resnet50_dcn

2 主干网络的设计范式

2.1 卷积层

2.1.1使用深度可分离卷积

较好的应用就是Xception,Xception在PaddlePaddle的排行榜中也展现了很好的结果;

2.2 特征融合——使用1x1卷积

使用1x1卷积来进行特征融合;

2.3 激活函数——ReLU(推荐)

可以选择的激活函数有ReLU和Mish函数;

2.3.1Mish函数

这里记录一个Mish函数的简单实现,

class Mish(nn.Module):def forward(self, x):return x * torch.nn.functional.softplus(x).tanh()

实际测试的结果不是很理想,原因是以上用pure-torch实现的Mish()函数会占用较多的显存,

(所以无奈之下,南溪只好将bs=3),不过这样子在QMV上的结果没有基于ReLU的resnet50_dcn效果好,

所以南溪还是暂时选择ReLU函数;

如果觉得《《南溪的目标检测学习笔记》——主干网络backbone设计的学习笔记》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。