本文主要是介绍YOLOV5 改进:替换backbone为EfficientNet,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
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_
这篇关于YOLOV5 改进:替换backbone为EfficientNet的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!