timm documentation

インストール

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

インストール

開始する前に、環境を設定し、適切なパッケージをインストールする必要があります。timmにはPython 3.10以降が必要です。

仮想環境

依存関係の競合を避け、環境を整理するために、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

代わりに、timmをGitHubから直接インストールすることで、最新のバージョンを取得できます:

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 で利用可能な事前学習済みモデルのうち最初の5つをリスト表示します(アルファベット順にソートされています)。以下の出力が表示されるはずです:

['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])"
Update on GitHub