- 博客(2)
- 收藏
- 关注
RSS订阅原创 yolov8添加注意力机制三部曲：添加localwindowsattention注意力
瓶颈模块中添加，改ultralytics-cfg-models-v8-yolov8-test-yaml(从yolov8改进，如果从其他模型要做出相应修改)最先使用的BIformer，but，显存炸了，搜索发现BI方法就是存在内存占用大计算代价高的特点，所以选了localwindowsattention。然后，在代码最下方添加注意力代码，源码放在下方，注意：conv2d_BN也需要。调用方法代码位置nn-modules-tasks.py，如下图所示，观察训练过程，如下图所示，就代表添加成功啦！
2024-04-09 21:14:06
1205
1
原创 Collecting package metadata (current_repodata.json): failed（解决方案*3）
报错结果：解决方案1：1.打开，输入下面命令手动删除.condarc。
2024-03-14 21:07:34
247
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人