金子邦彦研究室人工知能Windows で動く人工知能関係 Pythonアプリケーション,オープンソースソフトウエア)マルチオブジェクトトラッキング(MOT)と人物再識別(BoT-SORT,Python 3.9,PyTorch 1.14を使用)(Windows 上)

マルチオブジェクトトラッキング(MOT)と人物再識別(BoT-SORT,Python 3.9,PyTorch 1.14を使用)(Windows 上)

前準備

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

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

サイト内の関連ページ

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

関連する外部ページ

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

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

BoT-SORT の動作のために Python 3.9, 3.8, 3.7, 3.6 のいずれかの利用をおすすめする.

サイト内の関連ページ

関連する外部ページ

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 のインストールを行う.

関連する外部ページ

BoT-SORT のインストール(Windows 上)

PyTorch 1.13 , numpy 1.19.5のインストール(Windows 上)

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

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

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

    python -m pip install -U --ignore-installed pip
    python -m pip install numpy==1.19.5 torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
    python -c "import torch; print(torch.__version__, torch.cuda.is_available())" 
    

    サイト内の関連ページ

    関連する外部ページ

cython, cocoapi, faiss のインストール(Windows 上)

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

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

  2. cython, cocoapi, faiss のインストール

    「python -m pip install -U faiss-gpu」でエラーメッセージが出たときは無視してください.

    python -m pip install -U cython
    python -m pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
    python -m pip install -U faiss-cpu
    python -m pip install -U faiss-gpu
    

BoT-SORT のインストール(Windows 上)

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

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

  2. BoT-SORT のダウンロードと前準備

    cd %HOMEPATH%
    rmdir /s /q BoT-SORT
    git clone https://github.com/NirAharon/BoT-SORT.git
    python -m pip install -U pip
    python -m pip install -U numpy==1.19.5 opencv-python loguru scikit-image scikit-learn tqdm Pillow thop ninja tabulate motmetrics filterpy h5py matplotlib scipy prettytable easydict tensorboard pyyaml yacs termcolor gdown pandas onnx onnxruntime onnx-simplifier 
    python -m pip install -U lap
    python -m pip install -U tensorboard
    cd BoT-SORT
    python -m pip install -r requirements.txt
    
  3. setup.py の書き換え

    cd %HOMEPATH%\BoT-SORT
    notepad setup.py 
    

    次のように,「with open("README.md", "r", encoding='utf-8') as f:」に書き換える.保存する.

    [image]
  4. インストール

    cd %HOMEPATH%\BoT-SORT
    python setup.py develop
    mkdir pretrained
    
  5. 学習済みモデル bytetrack_x_mot17.pth.tar のダウンロード

    次のページの指示に従う

    https://github.com/NirAharon/BoT-SORT

    このページの中の「ByteTrack」をクリック.

    [image]

    ダウンロードしたファイル bytetrack_x_mot17.pth.tar は %HOMEPATH%\BoT-SORT\pretrained に置く

    [image]

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

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

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

  2. cython_bbox のインストール

    cd %HOMEPATH%
    curl -O https://files.pythonhosted.org/packages/fa/b9/fc7d60e8c3b29cc0ff24a3bb3c4b7457e10b7610fbb2893741b623487b34/cython_bbox-0.1.3.tar.gz
    "c:\Program Files\7-Zip\7z.exe" x cython_bbox-0.1.3.tar.gz
    cd dist
    "c:\Program Files\7-Zip\7z.exe" x cython_bbox-0.1.3.tar
    notepad cython_bbox-0.1.3\setup.py
    

    次のように書き換えて,保存する.

    [image]

    引き続き次の操作を行う

    cd %HOMEPATH%\BoT-SORT
    cd dist
    cd cython_bbox-0.1.3
    python setup.py build
    python setup.py develop
    python setup.py install
    
  3. インストール終了の確認

    エラーメッセージが出ていないこと.

    [image]

MOT と人物再識別の実行

  1. 動画ファイルを%HOMEPATH%\BoT-SORT に置く
  2. 実行してみる.

    次のページの指示に従う

    https://github.com/NirAharon/BoT-SORT

    1-1.avi」のところには,処理したい動画ファイル名を指定する.

    cd %HOMEPATH%\BoT-SORT
    python tools/mc_demo.py video --path 1-1.avi -f yolox/exps/example/mot/yolox_x_mix_det.py -c pretrained/bytetrack_x_mot17.pth.tar --with-reid --fuse-score --fp16 --fuse --save_result
    
  3. 実行の終了の確認

    エラーメッセージが出ていないこと.

    結果が保存されるファイルとそのディレクトリが表示されるので確認する.

    [image]
  4. いま表示されたディレクトリに動画ファイルがある.

    [image]
  5. 動画ファイルを表示してみる.

    [image]

    [image]

    動画ファイル: 1-1.avi