金子邦彦研究室人工知能Windows で動く人工知能関係 Pythonアプリケーション,オープンソースソフトウエア)オブジェクト・トラッキングとセグメンテーション(Track Anything,Python,PyTorch を使用)(Windows 上)

オブジェクト・トラッキングとセグメンテーション(Track Anything,Python,PyTorch を使用)(Windows 上)

目次

  1. 前準備
  2. Track Anything のインストール(Windows 上)
  3. Track Anything の動作確認(Windows 上)

前準備

Git のインストール(Windows 上)

Gitは,バージョン管理システム.ソースコードの管理や複数人での共同に役立つ.

サイト内の関連ページ

Windows での Git のインストール: 別ページ »で説明している.

関連する外部ページ

Git の公式ページ: https://git-scm.com/

Python のインストール(Windows 上)

サイト内の関連ページ

関連する外部ページ

Python の公式ページ: https://www.python.org/

Build Tools for Visual Studio 2022,NVIDIA ドライバ,NVIDIA CUDA ツールキット 11.8,NVIDIA cuDNN 8.6 のインストール(Windows 上)

サイト内の関連ページ

NVIDIA グラフィックスボードを搭載しているパソコンの場合には, NVIDIA ドライバNVIDIA CUDA ツールキットNVIDIA cuDNN のインストールを行う.

関連する外部ページ

PyTorch のインストール(Windows 上)

  1. Windows で,コマンドプロンプト管理者として実行

    コマンドプロンプトを管理者として実行: 別ページ »で説明

  2. PyTorch のページを確認

    PyTorch のページ: https://pytorch.org/index.html

  3. 次のようなコマンドを実行(実行するコマンドは,PyTorch のページの表示されるコマンドを使う).

    次のコマンドを実行することにより, PyTorch 2.3 (NVIDIA CUDA 11.8 用)がインストールされる. 但し,Anaconda3を使いたい場合には別手順になる.

    事前に NVIDIA CUDA のバージョンを確認しておくこと(ここでは,NVIDIA CUDA ツールキット 11.8 が前もってインストール済みであるとする).

    PyTorch で,GPU が動作している場合には,「torch.cuda.is_available()」により,True が表示される.

    python -m pip install -U --ignore-installed pip
    python -m pip uninstall -y torch torchvision torchaudio
    python -m pip install -U torch torchvision torchaudio numpy --index-url https://download.pytorch.org/whl/cu118
    python -c "import torch; print(torch.__version__, torch.cuda.is_available())" 
    

    [image]

    Anaconda3を使いたい場合には, Anaconda プロンプト (Anaconda Prompt)管理者として実行し, 次のコマンドを実行する. (PyTorch と NVIDIA CUDA との連携がうまくいかない可能性があるため,Anaconda3を使わないことも検討して欲しい).

    conda install -y pytorch torchvision torchaudio pytorch-cuda=11.8 cudnn -c pytorch -c nvidia
    py -c "import torch; print(torch.__version__, torch.cuda.is_available())" 
    

    サイト内の関連ページ

    関連する外部ページ

Track Anything のインストール(Windows 上)

FFmpeg のインストール(Windows 上)

Windows での FFmpeg のインストール(Windows 上): 別ページ »で説明している

Track Anything のインストール(Windows 上)

  1. Windows で,コマンドプロンプト管理者として実行

    コマンドプロンプトを管理者として実行: 別ページ »で説明

  2. ダウンロードとインストール

    cd %HOMEPATH%
    rmdir /s /q Track-Anything
    git clone https://github.com/gaomingqi/Track-Anything
    cd Track-Anything
    python -m pip install -U pip
    python -m pip uninstall -y tensorflow tensorflow-cpu tensorflow-gpu tensorflow-intel
    python -m pip install -U tensorflow==2.10.1 keras onnx protobuf
    python -m pip install -r requirements.txt
    
  3. 終了の確認

    [image]
  4. app.py を編集する

    cd %HOMEPATH%\Track-Anything
    notepad app.py
    

    [image]
  5. app.py を次のように書き直して,保存する

    変更前: args.device = "cuda:3"

    変更後: args.device = "cuda:0"

    変更の結果,次のようになる.

    [image]

Track Anything の動作確認(Windows 上)

  1. Windows で,コマンドプロンプトを実行
  2. 次のコマンドを実行して,Track Anything を起動

    初回起動時は,ファイルのダウンロード,MMCV とMMEngine のインストールが行われる. 起動するまで少し時間がかかる.

    cd %HOMEPATH%\Track-Anything
    python app.py --device cuda:0
    

    メモリに余裕がないときは,メモリを節約できる次のオプションを検討する

    cd %HOMEPATH%\Track-Anything
    python app.py --device cuda:0 --sam_model_type vit_b
    
  3. 次の表示が出たときは「アクセスを許可する」をクリック.

    表示が出なくても問題ない.

    [image]
  4. 起動できたことを確認

    [image]
  5. Web ブラウザで http://127.0.0.1:12212/ を開く
  6. Web ブラウザでビデオファイルを選ぶか、アップロード

    [image]
  7. Get video info」 をクリック

    結果が表示されるまで待つ.

    [image]
  8. Web ブラウザの中の下の画面で、オブジェクトをクリックすると,セグメンテーションが行われる.

    これは,トラッキングしたいオブジェクトを選ぶ操作である.

    クリックは複数回行うことができる.

    「Clear clicks」でやり直すことができる.

    [image]
  9. Web ブラウザの中の右の画面で、Tracking をクリック。

    結果が表示されるまで待つ.

    [image]
  10. 結果は,右の画面で確認

    [image]

    結果を保存した動画ファイル:trackanything.mp4