本文给大家带来的改进机制是BiFPN双向特征金字塔网络,其是一种特征融合层的结构,也就是我们本文改进YOLOv5模型中的Neck部分,它的主要思想是通过多层级的特征金字塔和双向信息传递来提高精度。本文给大家带来的结构可以让大家自行调节网络结构大小,同时能够达到一定的轻量化效果(需要注意的是BiFPN正常是需要五个检测头的,...
第一步:损失函数定义到YOLOv7或者YOLOv5的models/common.py中。 classGSConv(nn.Module):# GSConv https://github.com/AlanLi1997/slim-neck-by-gsconvdef__init__(self,c1,c2,k=1,s=1,g=1,act=True):super().__init__()c_=c2//2self.cv1=Conv(c1,c_,k,s,None,g,act)self.cv2=Conv(...