使用 Timm 库替换 YOLOv8 主干网络 | 1000+ 主干融合YOLOv8
文章目录
- 前言
- 版本差异说明
- 替换方法
- parse_moedl( ) 方法
- _predict_once( ) 方法
- 修改 yaml ，加载主干
- 论文引用
timm 是一个包含最先进计算机视觉模型、层、工具、优化器、调度器、数据加载器、数据增强和训练/评估脚本的库。
该库内置了 700 多个预训练模型，并且设计灵活易用。(截止到本博客创作时间，已经 1000+ 模型了)
这个库也是 HuggingFace 团队的一个比较知名的库，如果大家将来想从事算法工作，这个会对你有很大的帮助。
代码库地址：