![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
yolov5
文章平均质量分 79
飞羽QQ
这个作者很懒,什么都没留下…
展开
-
关于yolov5训练后权重较大的原因及其解决方案
yolov5在官方预训练权重的基础之上再训练后权重较预训练权重大很多,为了大家更加直观的了解为什么及其解决方案,此篇简单介绍一下。精度变化官方给的预训练权重是FP16,而我们训练的时候是使用混合精度训练(支持CUDA才行),半精度训练只能在CUDA下进行,不支持CUDA默认是使用单精度训练,最终我们保存的权重是FP32,较FP16储存空间大了一倍。直接上代码视图:import argparsefrom models.common import *if __name__ == '__main原创 2020-08-22 16:19:21 · 13620 阅读 · 13 评论 -
yolov5训练详解
最近跑了一下yolov5,在此给大家介绍一下如何用yolov5训练自己的数据集,github官方代码https://github.com/ultralytics/yolov5官方权重我已经下载下来了,见https://download.csdn.net/download/qq_41334243/12572600【注】本文所说的训练只针对官方代码。1)数据集制作数据集选择:(1)自己做的数据集(2)开源数据集(1)自己做的数据集本人使用精灵标注助手做了一个猫狗识别,把每张图片上的猫和狗都框出来,原创 2020-07-18 13:21:45 · 32554 阅读 · 24 评论