- 博客(1)
- 收藏
- 关注
原创 YOLOv8添加注意力机制后可以从预训练模型开始训练
本人在YOLOv8中添加注意力机制发现新的模型,例如将yolov8s.yaml文件更改成为yolov8s-Attention.yaml文件,也就是在C2f里添加了注意力机制。再看看更改后的代码,发现第一轮的训练结果为0.832,说明预训练加载到模型当中了。这里我们发现第一轮的训练结果为0,说明是从头开始训练。发现注意力机制添加成功!
2023-09-08 08:35:07 2604 13
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人