timm 文件

安裝

Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

安裝

在開始之前,你需要設定你的環境並安裝適當的軟體包。timmPython 3+ 上經過測試。

虛擬環境

你應該將 timm 安裝在 虛擬環境 中,以保持整潔並避免依賴衝突。

  1. 建立並導航到你的專案目錄:

    mkdir ~/my-project
    cd ~/my-project
  2. 在你的目錄中啟動一個虛擬環境

    python -m venv .env
  3. 使用以下命令啟用和停用虛擬環境:

    # 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])"
< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.