1、介绍
本章将会把yolov5的主干网络替换成EfficientNet V2,这里直接粘贴代码
详细的可以参考之前的内容:YOLOV5 改进:替换backbone(MobileNet为例)_yolov5主干网络更换为mobile-CSDN博客
更多的backbone更换参考本专栏: YOLOV5 实战项目(训练、部署、改进等等)_听风吹等浪起的博客-CSDN博客
更换的顺序如下:
本文更换的是efficientNet V2
2、更改common 文件
将下面代码加入common最下面即可:
# EfficientNetV2
import torch
import torch.nn as nn
# 定义卷积-批归一化-激活函数模块
class stem(nn.Module):
def __init__(self, c1, c2, kernel_