YOLOV5 改进:更换主干网络为vgg16(包含迁移学习)

1、前言

本章将介绍如何将官方的定义的vgg16代码更换到yolov5的主干网络

在之前实现了vgg的更换代码,只不过上次的vgg是自己定义在py脚本中的,代码进行稍微更改才行。

这里官方实现的更为简单,而且相同网络结构的话,可以跟着本章自己更改

具体的还是参考这一章:YOLOV5 改进:替换backbone(MobileNet为例)-CSDN博客

Tips :之前在这篇博文中YOLOV5 改进:替换backbone为Vgg_yolov5替换vgg-CSDN博客

有个兄弟问可不可以将训练好的再次进行迁移学习,当然是可以的。

因为迁移学习只是将网络参数进行载入,其实就是参数初始化,既然网络是一样的,当然可以载入,和其他的没有关系。

至于yolov5s.pt 为什么每次都可以载入我们自定义的数据集,因为官方的代码对载入权重进行了判断,只会把相同的部分载入,不相同的丢弃即可,其实就是很简单的迁移学习策略

为了方便理解,本章将训练200个epoch,然后载入训

最近更新

  1. TCP协议是安全的吗?

    2024-03-29 06:52:07       16 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-03-29 06:52:07       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-03-29 06:52:07       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-03-29 06:52:07       18 阅读

热门阅读

  1. 【云开发笔记No.7】敏捷开发

    2024-03-29 06:52:07       18 阅读
  2. 再谈敏捷开发

    2024-03-29 06:52:07       19 阅读
  3. Linux centos7离线搭建FTP

    2024-03-29 06:52:07       16 阅读
  4. 数组合并小程序

    2024-03-29 06:52:07       15 阅读
  5. 【WPF应用21】WPF 中的 TextBox 控件详解与示例

    2024-03-29 06:52:07       17 阅读
  6. Superset二次开发之webpack.config.js 功能模块解读

    2024-03-29 06:52:07       19 阅读
  7. 大语言模型中的强化学习与迁移学习技术

    2024-03-29 06:52:07       16 阅读
  8. PyTorch中nn.ReLU函数说明

    2024-03-29 06:52:07       19 阅读
  9. flutter 打包成web应用后怎么通过url跳转页面

    2024-03-29 06:52:07       18 阅读
  10. Flutter——用户关闭键盘后强制拉起键盘

    2024-03-29 06:52:07       19 阅读
  11. 解决 docker swarm 节点容器无法获取客户端真实IP

    2024-03-29 06:52:07       19 阅读