YOLOV5 改进:替换backbone为Vgg

news/2024/7/10 0:45:48 标签: YOLO, 目标检测, 深度学习, 人工智能

1、前言

参考上一章的博文(YOLOV5 改进:替换backbone(MobileNet为例)-CSDN博客)将yolov5的backbone换成自定义的vgg网络

网络参数量很多,并且刚开始训练的时候精度很差,应该是没有迁移学习导致的。

大概经历了30-40多个epoch,网络才进行收敛, 加大epoch可以提升网络的精度。

不过由于vgg的参数量太多了,这里只训练100个epoch,感兴趣的可以自己尝试下

2、替换common文件

# 增加VGG 作为 backbone
class VGG(nn.Module):  # 定义VGG网络
    def __init__(self, features, num_classes=1000):  # num_classed 为分类的个数
        super(VGG, self).__init__()
        self.features = featu

http://www.niftyadmin.cn/n/5440355.html

相关文章

cmake 报错: could not open ‘kernel32.lib‘: no such file or directory

首次用hello world想配置cmake。powershell下执行cmake .. -G "MinGW Makefiles"报的这个错。 因为查到这个库是windows下的,于是想着换成Unix试下。 执行:cmake .. -G "Unix Makefiles" 成功 cmake 版本: cmake ver…

【图像分割】使用Otsu 算法及迭代计算最佳全局阈值估计并实现图像分割(代码实现与分析)

本实验要求理解全局阈值分割的概念,并实现文本图像分割。需要大家深入理解Ostu 算法的实现过程及其迭代原理,同时通过学习使用Otsu 算法及其迭代,实践图像分割技术在文本图像处理中的应用。 以下将从实验原理、实验实现、实验结果分析三部分对…

基于深度学习的场景文本检测

CTPN 简介: 基于目标检测方法的文本检测模型,在Faster RCNN的基础上进行了改进,并结合双向LSTM增强了序列提取特征,通过anchor和gt的设计将文本检测任务转化为一连串小尺度文本框的检测。 解决问题: 文本长短不一&…

python--剑指offer--题目目录-学习计划

第一天 09. 用两个栈实现队列30. 包含min函数的栈 第二天 06. 从尾到头打印链表24. 反转链表35. 复杂链表的复制 第三天 05. 替换空格58 - II. 左旋转字符串 第四天 03. 数组中重复的数字53 - I. 在排序数组中查找数字 I53 - II. 0~n-1中缺失的数字 第五天 04.…

课时66:流程控制_case条件控制_嵌套案例

2.3.4 嵌套案例 学习目标 这一节,我们从 案例需求、简单实践、小结 三个方面来学习。 案例需求 运维管理人员,通过监控平台获取站点的运行状态数据信息,当发现问题的时候,根据情况进行后续判断:状况1: …

机器视觉系统选型-相机基础知识

相机基础知识: 分辨率(Resolution):相机采集图像的像素点数 像素分辨率(mm/pixel):每个像素代表的毫米值 快门方式: 全局快门/全域快门(Global Shutter):让整个感光元器件每行像素全部在同一时间…

使用Python查找字符串中包含的多个元素

目录 一、引言 二、基本字符串操作 使用in关键字查找子字符串 使用循环和条件判断查找多个子字符串 三、使用正则表达式进行高级搜索 导入re模块 使用re.search()查找单个模式 使用re.findall()查找多个模式 四、案例与代码 五、优化与扩展 六、总结 一、引言 在Py…

C++进阶:二叉搜索树介绍、模拟实现(递归迭代两版本)及其应用

上次介绍完多态后:C进阶:详解多态(多态、虚函数、抽象类以及虚函数原理详解) 也是要开始继续学习了 文章目录 1.二叉搜索树1.1概念1.2二叉搜索树特性1.3 二叉搜索树的操作 2.模拟实现2.1项目文件规划2.2基本结构2.3各种接口、功能…