Timm使用教程

Timm使用教程

关注B站可以观看更多实战教学视频：hallo128的个人空间

参考：timm使用教程

kaggle代码链接：https://www.kaggle.com/code/hallo123/timm-tutorial（逐步调试运行）
官方指导链接：https://huggingface.co/docs/timm/quickstart#quickstart

timm（Pytorch Image Models）项目是一个图像分类模型库，通过timm可以轻松的搭建出各种sota模型（目前内置预训练模型592个，包含densenet系列、efficientnet系列、resnet系列、vit系列、vgg系列、inception系列、mobilenet系列、xcit系列等等），并进行迁移学习。

核心掌握点

1.安装timm
2.加载预训练模型
2.1.手动加载对应模型
2.2 计算模型参数量
3.查看模型列表
4.模型微调最后一层
5.特征提取
可视化网络结构
可视化模型结构
6.查看/设定模型的图像增强方法
7.使用预训练模型进行推断
（1）待预测的图片
（2）加载模型，获取数据预处理方法
（3） 调用模型预测
（4） 用softmax获取预测概率
（5）根据索引号对应分类
8.使用timm中可复用的模块
9.修改模型
9.1 修改模型层
9.2 添加外部输入
9.3 添加额外输出

使用教程：timm使用教程