timm documentation
インストール
はじめに
チュートリアル
timmのモデルAPIモデルの適応とエクスポート事前学習モデルを特徴抽出器として使用ハイパーパラメータ (HParams)公式トレーニングスクリプトの使用🤗 Hugging Face Hub からモデルを共有して読み込む
モデルページ
リファレンス
インストール
開始する前に、環境を設定し、適切なパッケージをインストールする必要があります。timmにはPython 3.10以降が必要です。
仮想環境
依存関係の競合を避け、環境を整理するために、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
代わりに、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])"