timm 文档
安装
加入 Hugging Face 社区
并获得增强的文档体验
开始使用
安装
在开始之前,你需要设置你的环境并安装适当的软件包。timm
在 Python 3+ 上经过测试。
虚拟环境
你应该将 timm
安装在 虚拟环境 中,以保持整洁并避免依赖冲突。
创建并导航到你的项目目录:
mkdir ~/my-project cd ~/my-project
在你的目录中启动一个虚拟环境
python -m venv .env
使用以下命令激活和停用虚拟环境:
# Activate the virtual environment source .env/bin/activate # Deactivate the virtual environment source .env/bin/deactivate
创建虚拟环境后,你可以在其中安装 timm
。
使用 pip
安装 timm
最直接的方法是使用 pip
pip install timm
或者,你可以直接从 GitHub 安装 timm
以获取最新的、前沿的版本
pip install git+https://github.com/rwightman/pytorch-image-models.git
运行以下命令来检查 timm
是否已正确安装
python -c "from timm import list_models; print(list_models(pretrained=True)[:5])"
该命令会列出 timm
中可用的前五个预训练模型(按字母顺序排序)。你应该看到以下输出
['adv_inception_v3', 'bat_resnext26ts', 'beit_base_patch16_224', 'beit_base_patch16_224_in22k', 'beit_base_patch16_384']
从源码安装
从源码构建 timm
可以让你对代码库进行更改。要从源码安装,请克隆仓库并使用以下命令进行安装
git clone https://github.com/rwightman/pytorch-image-models.git
cd pytorch-image-models
pip install -e .
同样,你可以使用以下命令检查 timm
是否已正确安装
python -c "from timm import list_models; print(list_models(pretrained=True)[:5])"