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

yolov5

【目标检测】-----YOLOV5网络结构详解_第1张图片
【目标检测】-----YOLOV5网络结构详解_第2张图片
【目标检测】-----YOLOV5网络结构详解_第3张图片

可以看出,相比于之前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

你可能感兴趣的:(目标检测,目标检测,深度学习,神经网络)