YOLOv5 COCO数据集训练 |简记

标签：论文代码环境搭建

使用的代码： YOLOv5 in PyTorch

该代码：自定义数据集官方教程

致敬大佬！

环境搭建

数据准备

COCO数据集下载
COCO数据集txt格式标签下载
通过 vim yolov5-master/data/scripts/get_coco.sh 可发现默认的数据路径安排如下：
通过 vim yolov5-master/data/coco.yaml 注释掉下载脚本：
最终的项目路径如下：

训练：

重新开始训练：

python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 32  --device  1,2 --epochs 300 --img 640

活力全开，时长统计

time python train.py --data coco.yaml --cfg yolov5s.yaml --weights '' --batch-size 64  --device  1,2 --epochs 10 --img 640

2块 RTX 5000 16G | --batch-size 64

5个 epoch 耗时 100分钟，1小时 3个 epoch ，那么训练该数据集全部数据，预估 300epoch 训练需要 100 小时；

本文链接：https://blog.csdn.net/sinat_28442665/article/details/108813518

智能推荐

python基础学习七——类

python 面向对象相对别的语言来说缺少两个功能： 1、python不具备重载，重载是指在同一个类中，使得方法有相同的名称，但是有不同的参数列表，但由于python函数具有强大的参数处理功能，因此这不是一个问题。 2、python不存在强制数据隐私的机制，不过若想创建属性（实例变量或方法）时在属性名前以两个下划线引导，python就会阻止无心的访问，因此可以认为是私有的。如果一个方法是预定义的...

手把手教你 vue3 写个简单的购物车DEMO

先上效果图：demo传送门 github 地址代码：...

JavaScript对象，JSON对象，JSON字符串之间的相互转化大全{包含三者的具体介绍和定义}

欢迎来到Altaba的博客 2017年12月22日 1、什么是json？ JSON(JavaScript Object Notation) 是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。 2、json的格式？ JSON建构于两种结构：（1）“名称/值&rdquo...

R语言3.12 对应分析

对应分析是主成分和因子分析的进一步扩展对应分析的提出是基于因子分析的不足，因子分析方法分为R型因子分析和Q型因子分析。R型因子分析研究变量（指标）之间的相关关系，Q型因子分析研究样本之间的相关关系。但有时不仅关心变量之间或样本之间的相关关系，还关心变量和样本之间的对应关系，这是因子分析不能解释的。对应分析的作用：分析两组或多组因素之间关系的有效方法，在离散情况下，建立因素间的列联表来对数据进行...

ZVulDrill靶场漏洞分析（简单的代码审计学习）

装了一个小靶场来进行简单的代码审计学习搜索框sql注入代码分析该语句没有任何防护，典型的搜索框注入搜索输入1后监控变量 sql语句 SELECT * FROM comment WHERE comment_text LIKE '%1%' 所以只要构造语句闭合%‘就能执行sql注入 Payload: %’ and 1=2 --+ And 1=1 页面正常，存在注入，不再...