PyTorch Image Models (timm) 图像模型库

提供大量图像模型和预训练权重,以及训练与评估脚本

需要技术AI 整理,待开发者认领计算机视觉
评分人数不足呦探分评分人数不足大众分评分方法
这个页面由 AI 根据官网公开信息整理,可能有错。产品团队可以认领后修改。 认领这个产品

这个项目只有源代码,需要会编程才能用。

星标
3.7万
GitHub 下载量
1085.6万
复刻
5223
主要语言
Python
许可证
Apache-2.0
最近更新
复制安装指令(给 AI 编程工具)

粘贴到 Codex、Claude Code、Cursor、Trae 等 AI 编程工具里,回车就会开始安装。第一次复制前请先阅读风险提示。

这些文件由项目作者放在 GitHub 上,本站不开发、不托管这些软件,也不检测其代码;国内访问 GitHub 可能较慢。使用前请阅读 第三方软件风险提示和项目的许可证。 数据更新于 2026/11/01 报告问题

介绍

PyTorch Image Models(简称 timm)是基于 PyTorch 的图像模型库,最初由 Ross Wightman 开发,文档和权重托管在 Hugging Face。它汇集了 ResNet、EfficientNet、ViT、ConvNeXt、Swin 等众多架构,大多附带预训练权重,常用作视觉任务的骨干网络。

能做什么

  • 用统一接口加载图像模型和预训练权重,替换分类头或提取多尺度特征
  • 使用内置的训练、验证和推理脚本,支持多 GPU 分布式与混合精度训练
  • 调用库中的优化器、数据增强和正则化方法

运行需要:Python 与 PyTorch 环境(pip 安装);预训练权重从 Hugging Face Hub 下载,训练建议使用 GPU。部分权重需遵守 ImageNet 数据集的许可条款。

这段介绍由 AI 根据项目的公开说明整理,可能有疏漏,以项目主页为准。

评测

0 条评测

还没有评测。用过的话,来写第一条吧。

写评测