pytorch-image-models

一行代码调用数百个预训练视觉骨干，省去复现选型成本

开源 🏷️完全免费 🔧 仅开源项目，无在线服务，需自行部署计算机视觉 🍴 复刻 5222 🐛 待处理 17 👀 关注 315

GitHub ↗

GitHub 星标 ★ 37262

维护状态活跃

是否开源开源

定价模式 🏷️完全免费开源免费，Apache-2.0 许可证

项目预览

pytorch-image-models 项目界面预览图 — 图片来源：GitHub 官方仓库名片

项目数据

分类计算机视觉

开发团队huggingface

GitHubhttps://github.com/huggingface/pytorch-image-models

定价模式🏷️完全免费

价格说明开源免费，Apache-2.0 许可证

访问状态🔧 仅开源项目，无在线服务，需自行部署

是否开源是

开源协议Apache-2.0

主要语言Python

技术栈/模型augmix,convnext,distributed-training,efficientnet,image-classification,imagenet,maxvit,mixnet,mobile-deep-learning,mobilenet-v2,mobilenetv3,nfnets,normalization-free-training,optimizer,pretrained-models,pretrained-weights,pytorch,randaugment,resnet,vision-transformer-models

GitHub 星标★ 37262

上线时间2019-02-02 00:00:00

代码更新2026-10-31

仓库体积33.0 MB

本站数据更新2026-11-03

维护状态活跃

收录时间2026-09-14

浏览次数1

项目介绍

pytorch-image-models（常被称为 timm）是 PyTorch 生态中规模最大的图像编码器与骨干网络集合，由 Hugging Face 维护。它把 ResNet、EfficientNet、Vision Transformer、ConvNeXt、Swin、MaxViT、MobileNet 系列等数百个模型统一到一致的接口下，解决研究者与工程师在选型、复现和迁移时反复造轮子的问题。核心能力包括：一行代码创建任意模型并加载 ImageNet 预训练权重，支持特征提取、微调、验证与推理；提供训练、评估、导出（ONNX/TorchScript）脚本；支持分布式训练、混合精度、EMA、AugMix 等训练技巧；模型可无缝接入 Hugging Face Hub。凭借 3.7 万以上星标和持续更新，它已成为图像分类与视觉骨干网络的事实标准库，广泛用于论文复现、工业落地和教学。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化，观测越久曲线越完整

数据来源：GitHub API 逐日快照 · 收录于 2026-09-14 · 当前星标 ★ 37262

技术标签

distributed-training image-classification imagenet pretrained-models pytorch

使用教程

⚠ 本教程由 AI 依据该项目 README 整理；原始文档不完整，部分步骤为通用流程推断，实际安装请以项目官方文档为准。

难度：进阶约 10 分钟部署方式：库/依赖 4 步

⚠ 该项目 README 未提供完整安装说明，以下步骤为通用流程，请以官方文档为准。

环境要求

Python 环境（项目开发语言为 Python）
可访问 GitHub 与 Hugging Face（获取文档与预训练权重）
可联网浏览外部文档站点以完成安装与选型

安装与启动步骤

1查阅官方文档

README 未给出安装命令，官方安装与快速上手说明在 Hugging Face 的 timm 文档页，按该页指引操作。
2阅读实践指南

参考 Chris Hughes 的 Practitioner's Guide，或 timm.fast.ai（timmdocs），了解完整使用流程与常见用法。
3浏览可选模型

通过 paperswithcode 的 timm 页面按任务与结构浏览模型，确定需要的骨干网络后再进入安装环节。
4查看模型导出方案

README 的 Deployment 节指向 timmx，用于把 timm 模型导出为 ONNX、CoreML、LiteRT、TensorRT 等格式。

如何确认成功

README 未提供安装验证方式；请按官方文档 https://huggingface.co/docs/hub/timm 中的示例确认环境可用。

常见问题

Q：timm 的完整安装和上手说明在哪里？

A：官方文档地址为 https://huggingface.co/docs/hub/timm，README 的 Getting Started 一节即指向该文档，README 节选中未包含安装命令。

Q：有没有更详细的实战教程？

A：可参考 Chris Hughes 的《Getting Started with PyTorch Image Models (timm): A Practitioner's Guide》，以及 Aman Arora 维护的 timmdocs（timm.fast.ai）。

Q：如何挑选 timm 里的模型？

A：可访问 paperswithcode.com/lib/timm 浏览 timm 中包含的模型，再结合自己的任务选择对应骨干网络。

Q：如何把模型导出到其他推理格式？

A：README 的 Deployment 小节推荐 timmx（https://github.com/Boulaouaney/timmx），支持导出 ONNX、CoreML、LiteRT、TensorRT 等。

Q：最近新增了哪些模型？

A：README 的 What's New 列出 Qwen3-VL、Sapiens2 ViT、iFormer、EfficientViM、DeepSeek-V4 ViT、LowFormer 等新增模型与权重。

注意事项

README 节选未包含 pip/conda 等安装命令，安装方式请以 https://huggingface.co/docs/hub/timm 为准，不要凭猜测执行命令。
训练、验证、推理脚本在 README 的 Train, Validation, Inference Scripts 章节，节选中未展开具体参数。
预训练权重与文档托管在 Hugging Face，首次加载模型通常需要联网。