YOLOv8创新魔改教程(二)如何添加注意力机制

news/2024/7/10 23:32:34 标签: YOLO, 注意力机制

YOLOv8_0">YOLOv8创新魔改教程(二)如何添加注意力机制

(一)找代码

github找各种注意力机制的代码

(二)融合

1.创建文件

在ultralytics/nn/attention.py创建attention.py 文件

在这里插入图片描述
将找到的代码粘贴进来
在这里插入图片描述

2.修改task.py文件

打开 ultralytics/nn/tasks.py ,将注意力机制导入,名称就是注意力机制类的名称

在这里插入图片描述
在这里插入图片描述

在tasks.py文件中def parse_model函数,加上这段代码(需要接受通道数的加,不需要的不用加)

在这里插入图片描述

3.修改tyaml文件

创建一个yolov8-attention.yaml文件,将注意力机制添加上,注意修改网络层数

在这里插入图片描述

(三)运行

在这里插入图片描述

运行成功!


http://www.niftyadmin.cn/n/5236736.html

相关文章

智能手机的分层架构

一、用户界面(UI) 在智能手机架构中,用户界面(User Interface,简称 UI)是用户与应用程序进行交互的媒介。UI可以视为应用层的一个重要部分,它包括所有可视化的元素和用户交互的组件。 以下是UI…

使用Draw.io制作泳道图

使用Draw.io制作泳道图 一、横向泳道图1. 有标题泳道图2. 无标题泳道图3. 横纵向扩展泳道 二、纵向泳道图三、横纵交错地泳道图想做这样的图具体步骤1. 拖拽一个带标题的横向泳道图2. 拖拽一个带标题的单一图,并且把它放进Lane1中3. 其他注意 四、下载文件说明 一、…

观察者模式(常用)

观察者模式(Observer模式) 在现实世界中,许多对象并不是独立存在的,其中一个对象的行为发生改变可能会导致一个或者多个其他对象的行为也发生改变。例如,某种商品的物价上涨时会导致部分商家高兴,而消费者…

设置随机种子保证网络可复现性

原文链接:https://zhuanlan.zhihu.com/p/141063432 为什么使用相同的网络结构,跑出来的效果完全不同,用的学习率,迭代次数,batch size 都是一样?固定随机数种子是非常重要的。但是如果你使用的是PyTorch等…

LeetCode 2661. 找出叠涂元素:多次映射

【LetMeFly】2661.找出叠涂元素:多次映射 力扣题目链接:https://leetcode.cn/problems/first-completely-painted-row-or-column/ 给你一个下标从 0 开始的整数数组 arr 和一个 m x n 的整数 矩阵 mat 。arr 和 mat 都包含范围 [1,m * n] 内…

《我想假如在这里度过大学生活》火了,不愧是四川最美校园

近日,网上一篇关于成都理工大学校园环境的《我想假如在这里度过大学生活》火了。文章中的提到的大学环境优美,诗意盎然。一则则假如,带我们领略了校园的自然之美,发展之美、同时也感受到了大学时代的美好。 美丽的图书馆、阳光明媚…

mac上beyond compare提醒过期的处理方法

beyond comare 4官网本来就有免费版本,只是一些功能有限制,但是对于一般的文件比较,文件夹比较都有支持,所以也够用了。 但是在mac端有时候出现还是要输入序列号的情况,估计是bug。 解决方法是: 删除regis…

机器学习笔记 - 异常检测之OneClass SVM算法简述

一、异常检测是什么? 如下图,理想中我们可以找到一个框住大部分正常样本的决策边界,而在边界外部的数据点(蓝点)即视为异常。 但实际情况下数据都没有标签,因此很难定义正常还是不正常。异常检测的主要挑战如下:正常与异常行为之间的界限往往并不明确、不同的应…