自己在YOLOv8增加注意力模块遇到报错:KeyError
解决方案
目前优快云有一篇博文:彻底解决YOLOv8训练中的KeyError问题已经写的很好了,我在这里补充一种比较弱智的情况:
我个人是备份了原来的tasks.py为tasks_backup.py,在毫不知情的情况下把它和新写的tasks.py、attention.py等放在一起了,于是一直运行遇到报错,尝试了上文提到的所有方法都不好用。
结果发现是PyCharm Professional自动帮我把所有文件中含tasks的内容全部自动替换为了tasks_backup。我真的谢谢PyCharm。
解决方案是Ctrl Shift F查找所有文件里有tasks_backup的文本内容,将其替换为tasks。
或者直接将tasks_backup和tasks两个文件名分别修改为对方的文件名亦可。