【目标检测】-----YOLOV5网络结构详解

yolov5

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

可以看出,相比于之前v5.x,最新版的v6.x网络结构更加精简(以提高速度和推理性能),主要有以下更新:

<1>.Conv(k=6, s=2, p=2) 替换Focus
<2>.SPPF代替SPP,并且将SPPF放在主干最后一层
<3>.主干中的C3层重复次数从9次减小到6次
<4>.主干中最后一个C3层引入shortcut

1. CBL模块

网络中的标准卷积层,有2D卷积+BN层+激活函数(SiLU)组成,在之后的Bottleneck、C3、SPPF等结构中都会被调用。

import torch
import torch.nn as nn

版权声明:本文为m0_47405013原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。