【yolov5 v6.0】中断以后重新训练,增加epochs,冻结层
中断以后重新训练
有个resume的参数,将default从False改成True,然后他就会自己去找最新的权重继续训练了。
然后有个需要注意的点就是,不要为了想要备份最新的权重,然后把它复制一份出来,这样子,好像会造成路径的错乱,然后报错了很久哈哈哈(最后把复制出来的weights的文件夹删掉才好的)
增加训练的epochs
就是训练完了以后发现还没有收敛,想要继续增加epochs。我做的改动
- 修改resume参数
- 把epochs的参数修改成你需要的次数
- 在train文件夹里面加入语句,ckpt[‘epoch’] = 上一次训练的最后一个的次数
如果没有改的话可能会报错,大概的意思是说已经训练完了,不需要再继续训练了。 - 我还改了/runs/exp3中的opt.yaml文件中的epochs的个数,修改成了300,但是不知道这个有没有用处
训练的结果
tensorboard的使用
- 先在对应的环境下安装这个库
- 然后输入tensorboard --logdir=【event文件所在的文件夹路径】
- 然后在对应的端口打开
- 如果打开网页以后没有响应,或者找不到网页的话,那么就需要在xshell中把对应的端口弄到本地服务器上面。
- 然后再打开就好了
- 最后我的训练结果如上,虽然loss会突然增加,但是最后的趋势好像也和之前是一致的,初步判断,这样子增加训练的epochs是可行的。
冻结层
主要是对应一个freeze参数,这个参数默认是0,也就是不冻结的意思
对应的freeze的层数的代码主要是上面那样子的
freeze是一个列表,如果列表的长度大于1的话,那么freeze就是对应列表里面的数字的层数
例如输入的freeze为[0,9],那么对应的freeze=['model.0,model.9] 两层
如果输入的列表里面只有一个数字的话,那么对应的就是range(freeze[0])的层,如果输入的是10的话,那么对应的就是前10层['model.0, …, model.9] ,如果是默认输入的0的话,对应的freeze=[],没有需要冻结的层
那么对应的,可以考虑先冻结一部分层,再在这个基础上对后面的层进行修改的话,应该可以结合上面说到的几种方法进行测试?
【yolov5 v6.0】中断以后重新训练,增加epochs,冻结层相关推荐
- Yolov5 (v6.1)添加注意力机制
Apply Transformer in the backbone 1.要把注意力结构代码放到common.py文件中 2.手把手带你Yolov5 (v6.1)添加注意力机制(一)(并附上30多种顶会 ...
- 驱动人生6网卡版 v6.0.15.82 官方版
驱动人生6网卡版 v6.0.15.82 官方版 软件大小:115MB 软件语言:简体中文 软件性质:常用软件 软件授权:官方版 更新时间:2014-07-11 应用平台:/Win8/Win7/WinX ...
- 【目标检测】yolo系列:从yolov1到yolov5之YOLOv5训练自己数据集(v6.0)
一.源码下载及requirments 源码下载地址:https://github.com/ultralytics/yolov5 (持续更新中) 本人所用环境如下: pytorch:1.8(因为cuda ...
- 目标检测 YOLOv5网络v6 0版本总结
目标检测 YOLOv5网络v6.0版本总结 YOLOv5对比YOLOv4 输入端:在模型训练阶段,提出了Mosaic数据增强.自适应锚框计算.自适应图片缩放等: Backbone网络:融合其它检测算法 ...
- 目标检测 YOLOv5 - YOLOv5:v6版本多机多卡训练出现的错误及解决方案
2021年4月份发布的 YOLOv5:v5版本,2021年10月份发布的 YOLOv5:v6版本 发布了更小的Nano模型YOLOv5n和YOLOv5n6 一. 改进的方面 整合了 Roboflow, ...
- 目标检测 YOLOv5 - v6.2版本模型在瑞芯微 Rockchip设备从训练到C++部署实践
目标检测 YOLOv5 - v6.2版本模型在瑞芯微 Rockchip设备从训练到C++部署实践 flyfish 源码地址 https://github.com/shaoshengsong/rockc ...
- yolov5 v3.0训练出现KeyError错误
欢迎大家关注笔者,你的关注是我持续更博的最大动力 原创文章,转载告知,盗版必究 yolov5 v3.0训练出现KeyError错误 文章目录: 1 错误原因分析 2 错误解决方式 1 错误原因分析 出 ...
- yolov5的3.0版本代码在训练的时候报错:ImportError: cannot import name ‘amp‘ from ‘torch.cuda‘ 以及yolov5的3.0环境安装
欢迎大家关注笔者,你的关注是我持续更博的最大动力 原创文章,转载告知,盗版必究 yolov5的3.0版本代码在训练的时候报错:ImportError: cannot import name 'amp' ...
- yolov5 v3.0训练报错: torch.nn.modules.module.ModuleAttributeError: ‘BatchNorm2d‘ object has no attribute
欢迎大家关注笔者,你的关注是我持续更博的最大动力 原创文章,转载告知,盗版必究 yolov5 v3.0版本训练报错:torch.nn.modules.module.ModuleAttributeErr ...
最新文章
- rust怎么传送坐标_梦幻西游电脑版:金银锦盒修改后资金怎么攥?仅需知道六个小技巧!...
- sphinx 全文搜索引擎
- CCF NOI1014 写评语
- C++golden section search黄金分割搜索的实现算法(附完整源码)
- js数组的map方法以及parseInt方法
- 为什么Docker容器将占领世界
- oracle,EBS的库存数据视图
- 常用的、现成的!正则表达式及基础知识
- 多线程下单例模式:懒加载(延迟加载)和即时加载
- CCF201809-2 买菜(100分)【序列处理+差分】
- python - PyDev统一编码
- VB语言通用基础语句
- 【产品】 产品设计:ID设计和MD设计详解
- struct 结构体 所占字节数的计算
- 【读书笔记】巴比伦富翁的理财课
- 2021-2027全球与中国双断面隔离器市场现状及未来发展趋势
- 解析函数论 Page 22 达朗贝尔判别法 VS 柯西判别法
- Excel用正则表达式提取出输入正确的身份证号
- 零基础新手小白学编程必会的100个代码
- DEJA_VU3D - Cesium功能集 之 083-Cesium热力图实现完整版
热门文章
- VSCode查看gltf文件(glTF Tools插件)
- 网络管理维护(SNMP、RMON)期末考试大纲总结
- 苹果确认了!iPhone 将改用 USB-C 接口
- Python程序员爬出百套美女写真集,同样是爬虫,他为何如此突出?
- 错误:安装autocad2012的时候提示Command line option syntax error.Type Command/?for help.
- 学籍信息管理系统-------具体设计
- 一键绕过App签名验证
- stackoverflow热门问题(二)- 如何确定C的数组的大小
- 企业微信群机器人是什么?企微机器人如何自动发消息?
- 分列:将excel单元格的内容拆分为两列