本文主要是介绍YOLOV5 改进:更换主干网络为Resnet,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1、前言
之前实现了yolov5更换主干网络为MobileNet和vgg网络
本章将继续将yolov5代码进行更改,通过引用官方实现的resnet网络,替换原有的yolov5主干网络
替换的效果如下:
2、resnet 网络结构
测试的代码为官方的resnet34
通过summary 打印的resnet网络结构如下
==========================================================================================
Layer (type:depth-idx) Output Shape Param #
=============
这篇关于YOLOV5 改进:更换主干网络为Resnet的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!