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])"