YOLOv5，YOLOv8替换激活函数

目录

YOLOv5

YOLOv8

相关资料参考

---

YOLOv5

YOLOv5的激活函数就修改成功了

YOLOv8

也可以参考这个

YOLOv8修改激活函数_yolov8激活函数-CSDN博客

1、打开conv.py,将nn.SiLU改成你希望的   可以按鼠标滚轮查看和写入新的激活函数

class Conv(nn.Module):"""Standard convolution with args(ch_in, ch_out, kernel, stride, padding, groups, dilation, activation)."""default_act = nn.ReLU()  # default activation

2、打开block.py将nn.relu改成你希望的  同上

class BottleneckCSP(nn.Module):"""CSP Bottleneck https://github.com/WongKinYiu/CrossStagePartialNetworks."""def __init__(self, c1, c2, n=1, shortcut=True, g=1, e=0.5):  # ch_in, ch_out, number, shortcut, groups, expansionsuper().__init__()c_ = int(c2 * e)  # hidden channelsself.cv1 = Conv(c1, c_, 1, 1)self.cv2 = nn.Conv2d(c1, c_, 1, 1, bias=False)self.cv3 = nn.Conv2d(c_, c_, 1, 1, bias=False)self.cv4 = Conv(2 * c_, c2, 1, 1)self.bn = nn.BatchNorm2d(2 * c_)  # applied to cat(cv2, cv3)self.act = nn.ReLU()self.m = nn.Sequential(*(Bottleneck(c_, c_, shortcut, g, e=1.0) for _ in range(n)))

YOLOv8的激活函数就修改成功了

相关资料参考

YOLOv5改进 | 激活函数 | 十余种常见的激活函数一键替换【完整代码】_yolov5激活函数-CSDN博客