【记录解决YOLOv5加载权重文件报错问题】
报错原因:YOLOv5新旧版本不兼容,models下的yolo.py文件缺少DetectionModel模块代码。
解决步骤:
- 1.GitHub(链接: link)找到更新后的v5版本。
文章来源地址https://www.toymoban.com/news/detail-528313.html
- 2.找到models文件夹下的yolo.py,查找DetectionModel模块,将图片中的代码(Segment部分、BaseModel部分、DetectionModel部分)粘贴至报错的yolo.py里。
- 3.粘贴完后,运行yolo.py文件
文章来源:https://www.toymoban.com/news/detail-528313.html
到了这里,关于[记录解决YOLOv5加载权重文件报错问题]AttributeError: Can‘t get attribute ‘DetectionModel‘ on <module ‘models.yolo‘的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!