当前位置:网站首页>YoloV6:YoloV6+Win10---训练自己得数据集
YoloV6:YoloV6+Win10---训练自己得数据集
2022-07-07 17:53:00 【Mr曲末寒】
yolov6刚出来得那会我也跑了下,问题实在太多了,各种得bug,就暂时停了会,后面有时间了在看看就重新git下跑了跑。据说YoloV7也出来是AB大神团队出品得,不是某个人所谓得YoloV7....
1、数据集
我用的是自己得数据集,先看下数据集格式吧。
val和train存放得是图片,labels存放的是train val文件夹,里面放的是txt
同时在查看下配置文件,我的数据集 只有两个类别
此处还要改下dataset.py里面得一处;
2、训练
训练命令:
python tools/train.py --batch 6 --conf configs/yolov6s_finetune.py --data data/MyDataCoCo.yaml --device 0 --epochs 800 --workers 0
我是在自己本机训练的,所以bs设置得小点
训练结束,大家看到,map不为0了,那是我们使用得是yolov6s_finetune.py 如果大家使用的是yolov6s.py,相当于没有加载与训练模型,具体大家可以去Yolov6的github issues去看看。
https://github.com/meituan/YOLOv6/issues/205#issuecomment-1173409870
800轮训练时间将近19个小时。。。。。。。。。
3、测试
测试代码:
python tools/infer.py --weights runs/train/exp/weights/best_ckpt.pt --source CiWaData/val --yaml data/MyDataCoCo.yaml
测试结果:
4、ONNX导出
导出命令:
export_onnx.py --weights runs/train/exp/weights/best_ckpt.pt --img 640 --batch 1
边栏推荐
- Kunpeng developer summit 2022 | Kirin Xin'an and Kunpeng jointly build a new ecosystem of computing industry
- 剑指 Offer II 013. 二维子矩阵的和
- Throughput
- Leetcode force buckle (Sword finger offer 36-39) 36 Binary search tree and bidirectional linked list 37 Serialize binary tree 38 Arrangement of strings 39 Numbers that appear more than half of the tim
- 841. String hash
- 九章云极DataCanvas公司获评36氪「最受投资人关注的硬核科技企业」
- RESTAPI 版本控制策略【eolink 翻译】
- Ucloud is a basic cloud computing service provider
- ant desgin 多选
- Notes...
猜你喜欢
LeetCode_7_5
PMP每日一练 | 考试不迷路-7.7
Flink并行度和Slot详解
索引总结(突击版本)
位运算介绍
Jürgen Schmidhuber回顾LSTM论文等发表25周年:Long Short-Term Memory. All computable metaverses. Hierarchical reinforcement learning (RL). Meta-RL. Abstractions in generative adversarial RL. Soccer learn
Kirin Xin'an won the bid for the new generation dispatching project of State Grid!
648. 单词替换
Cloud 组件发展升级
Implement secondary index with Gaussian redis
随机推荐
一张图深入的理解FP/FN/Precision/Recall
LeetCode 648(C#)
【STL】vector
歌单11111
Boot 和 Cloud 的版本选型
9 atomic operation class 18 Rohan enhancement
R语言使用ggplot2函数可视化需要构建泊松回归模型的计数目标变量的直方图分布并分析构建泊松回归模型的可行性
Kirin Xin'an with heterogeneous integration cloud financial information and innovation solutions appeared at the 15th Hunan Financial Technology Exchange Conference
PMP对工作有益吗?怎么选择靠谱平台让备考更省心省力!!!
R语言ggplot2可视化:使用ggpubr包的ggviolin函数可视化小提琴图、设置palette参数自定义不同水平小提琴图的填充色、add参数在小提琴图添加箱图
How to open an account for stock speculation? Excuse me, is it safe to open a stock account by mobile phone?
Make insurance more "safe"! Kirin Xin'an one cloud multi-core cloud desktop won the bid of China Life Insurance, helping the innovation and development of financial and insurance information technolog
关于ssh登录时卡顿30s左右的问题调试处理
Tp6 realize Commission ranking
Openeuler prize catching activities, to participate in?
R语言ggplot2可视化:使用ggpubr包的ggstripchart函数可视化分组点状条带图(dot strip plot)、设置position参数配置不同分组数据点的分离程度
吞吐量Throughout
IP 工具类
UCloud是基础云计算服务提供商
841. 字符串哈希