使用 Timm 库替换 YOLOv8 主干网络 | 1000+ 主干融合YOLOv8

文章目录

    • 前言
    • 版本差异说明
    • 替换方法
      • parse_moedl( ) 方法
      • _predict_once( ) 方法
      • 修改 yaml ,加载主干
    • 论文引用


使用 Timm 库替换 YOLOv8 主干网络 | 1000+ 主干融合YOLOv8_第1张图片

timm 是一个包含最先进计算机视觉模型、层、工具、优化器、调度器、数据加载器、数据增强和训练/评估脚本的库。

该库内置了 700 多个预训练模型,并且设计灵活易用。(截止到本博客创作时间,已经 1000+ 模型了)

这个库也是 HuggingFace 团队的一个比较知名的库,如果大家将来想从事算法工作,这个会对你有很大的帮助。

代码库地址:

你可能感兴趣的:(YOLOv8改进实战,YOLO,计算机视觉,人工智能,timm,YOLOv8,改进,论文)