赞
踩
修改注意力机制,在task.py中import和parse_model中添加后,仍然出现keyerror 问题
可能是Yolov8版本的bug,解决办法:
anaconda3/envs/虚拟环境/lib/site-packages/ultralytics/nn/task.py
添加同样import 和pare_model,并将注意力机制函数文件复制到该目录下。
以GAM为例:我的目录D:\anaconda3\envs\yolov8\Lib\site-packages\ultralytics\nn
将gam.py文件复制到该目录下
task.py文件import添加GAM_Attention
prase_model添加
重新运行 yolo cfg=ultralytics\yolo\cfg\default.yaml
可以训练,结构已经修改,GAM已经在函数结构中:在2023.7月新yolov8新的更新,官方已经修复了此BUG ,直接用新版本跑增强改进训练不会出现这种问题。如果用了此版本之前的Yolov8注意力机制改进可能会出现这种问题,欢迎交流。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。