YOLOv5介绍
YOLOv5为兼顾速度与性能的目标检测算法。笔者将在近期更新一系列YOLOv5的代码导读博客。YOLOv5为2021.1.5日发布的4.0版本。
YOLOv5开源项目github网址
源代码导读汇总网址
本博客导读的代码为utils文件夹下的activations.py最后更新日期为2021.1.7
激活函数-数学表达
首先介绍YOLOv5激活函数中涉及到的函数其具体形式:
Sigmoid激活函数:
S i g m o i d ( x ) = 1 1 + e − x = e x e x + 1 Sigmoid(x) = \frac{1}{1+e^{-x} } = \frac{e^{x}}{e^{x}+1 } Sigmoid(x)=1+e−x1=ex+1ex
Swish激活函数:
S w i s h ( x ) = x 1 + e − β x Swish(x) = \frac{x}{1+e^{-\beta x}} Swish(x)=1+e−βxx
ReLU激活函数:
R e L U ( x ) = { x , if x > 0 0 , if x ≤ 0 = m a x ( 0 , x ) ReLU(x)= \begin{cases} x, & \text {if $x$ $>$ 0} \\ 0, & \text{if $x$ $\leq$ 0} \end{cases} = max(0,x) ReLU(x)={
x,0,if x > 0if x ≤ 0=max(0,x)
ReLU6限制上述函数输出在6以内:
R e L U 6 ( x ) = m i n ( 6 , m a x ( 0 , x ) ) ReLU6(x) = min(6,max(0,x)) ReLU6(x)=min(6,max(0,x))
Hard-swish激活函数:
H a r d − s w i s h ( x ) = x ∗ R e L U 6 ( x + 3 ) 6 Hard-swish(x) = \frac{x \ast ReLU6(x+3)}{6 } Hard−swish(x)