关于Ultra库改进后运行导入不生效问题记录与解决记录

一、问题描述

KeyError: 'FR_Head_Detect'

二、问题解决

2.1 解决思路一

每次改进后进入conda进行文件替换

         按照报错提示找到环境中加载的conda环境中对应的包内代码位置进行更换（此处因本人服务器正在疯狂训练中，故此懒得在新设备上再布置一次代码环境还原错误并修改，后续会补上步骤）

2.2解决思路二（建议）

每次训练使用实时的Ultralytics包

from ultralytics import YOLO
 
if __name__ == '__main__':
    # Load a model
    # #使用yaml配置文件来创建模型，并导入预训练权重，训练
    model = YOLO('ultralytics/cfg/models/yolov8-ghost-frm.yaml')  # 载入改进后模型配置
    model.load('yolov8n.pt')
    model.train(**{'cfg':'ultralytics/cfg/default.yaml', 'data':'ultralytics/cfg/datasets/VOC.yaml'}, epochs=300, imgsz=640, batch=32, name='ghost_frm')  # name：是此次训练结果保存的文件夹
                                                                                
    
    # 模型验证
    model = YOLO('runs/detect/train/weights/best.pt')
    model.val(**{'data':'ultralytics/cfg/datasets/VOC.yaml', 'name':'val_frm', 'batch':32})