YOLOv8 | 添加模块导致报错KeyError：已解决，详细步骤

目录

添加注意力机制等导致报错

报错的原因

解决错误流程

注意事项

代码分享

---

⭐欢迎大家订阅我的专栏一起学习⭐

🚀🚀🚀订阅专栏，更新及时查看不迷路🚀🚀🚀

💡魔改网络、复现论文、优化创新💡

因为yolov8封装的很好，只需要配置环境，不用看源码就可以直接进行训练，但是这也导致了一些问题，比如添加注意力机制的时候会导致一些keyerror的错误。下面我我们就详细说一下这个错误应该怎么解决。

添加注意力机制等导致报错

以添加GAM注意力机制为例子进行讲解

在添加注意力机制的代码，执行启动命令后

yolo detect train model=yolov8_GAM.yaml

会出现如下报错

Traceback (most recent call last):
  File "/usr/local/bin/yolo", line 8, in <module>
    sys.exit(entrypoint())
  File "/usr/l