目录
yolov5网络的主干是cspdarknet53,主干结构的作用就是用来提取特征的
YOLOv5的主干结构有CBS、C3和SPPF三种结构
一、CBS结构
代码
def autopad(k, p=None): # kernel, padding
# Pad to 'same'
if p is None:
p = k // 2 if isinstance(k, int) else [x // 2 for x in k] # auto-pad
return p
class Conv(nn.Module):
# Standard convolution
def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True): # ch_in, ch_out, kernel, stride, padding, groups
super().__init__()