【已解决】YOLOv8及YOLOv10中修改代码ultralytics包keyerror报错
修改代码或直接运行之所以会报错,是因为YOLOv8以及YOLOv10(使用了YOLOv8的项目结构)的项目结构中的ultralytics文件夹里面的路径的代码命名与pip install ultralytics这个第三方的包重复了,在终端使用yolo的命令训练的话,会默认解析pip下载的那个ultralytics包。
对于YOLOv8:如果你修改了里面的代码,比如添加自己的模块,然后运行,产生这个报错,有以下解决方法:
方法一(较为麻烦):把自己项目里面修改了的ultralytics文件夹的所有文件,例如task.py等文件,复制一份到pip安装的ultralytics里面进行覆盖,再次运行不会报错。
方法二(较为方便):
cd转到所在项目,先卸载ultralytics(如果是新环境,没有就不用管这条命令,直接运行下面的)
pip uninstall ultralytics
然后运行以下命令,使得ultralytics包与项目的ultralytics文件夹成为一体,在项目中更改比较方便。
pip install -e .
注意:如果YOLOv10与YOLOv8使用了不同环境,那就用上述方法即可。如果不是,往下看:
对于YOLOv10(与YOLov8使用一个环境的情况下):
由于其沿用了YOLOv8的结构,但是对task.py等多处文件进行了替换。
如果使用方法一进行直接替换,可能导致自己的同环境下的YOLOv8文件丢失,可以采用副本的方式,给task.py等需要更改的文件重命名,还未必能够替换成功,每次启用正确的task.py文件,另一个不能名字也为task.py,每次运行前进行替换。
如果使用方法二,便会直接把ultralytics又装回YOLOv10,这样更加麻烦,还可能导致环境出错。
综上所述,如果一个环境用于这两个YOLO,由于其特殊性,我认为不合理,因此建议多安装一个环境给YOLOv10使用,避免代码出错。
更多推荐
所有评论(0)