[记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘

【记录解决YOLOv5加载权重文件报错问题】

报错原因:YOLOv5新旧版本不兼容,models下的yolo.py文件缺少DetectionModel模块代码。

解决步骤:

  • 1.GitHub(链接: link)找到更新后的v5版本。

[记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘_第1张图片

  • 2.找到models文件夹下的yolo.py,查找DetectionModel模块,将图片中的代码(Segment部分、BaseModel部分、DetectionModel部分)粘贴至报错的yolo.py里。
    [记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘_第2张图片[记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘_第3张图片
    [记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘_第4张图片
  • 3.粘贴完后,运行yolo.py文件

你可能感兴趣的:(YOLO,python)