OpenCV-Object-Tracker-Sample

Python版OpenCVのTracking APIのサンプルです。
　

Requirement

• opencv-contrib-python 4.5.3.56 or later

Algorithm

2021/07/16時点でOpenCVには以下9アルゴリズムが実装されています。

• DaSiamRPN
• MIL
• GOTURN
• CSRT
• KCF
• Boosting(Legacy API)
• MOSSE(Legacy API)
• MedianFlow(Legacy API)
• TLD(Legacy API)

Usage

DaSiamRPNトラッカーのみのサンプルと、各アルゴリズムを比較するサンプルの2種類を用意しています。

DaSiamRPNトラッカーサンプル

以下コマンドでデモを起動してください。
最初のフレーム表示時にROI選択を行い追跡対象を指定します。
ESCキー押下でプログラム終了、スペースキー押下で追跡対象の再指定を行います。

python DaSiamRPN_sample.py

• --device
  動画ファイル、もしくはカメラデバイス番号の指定
  デフォルト："sample_movie/bird.mp4"
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540

各アルゴリズム比較サンプル

以下コマンドでデモを起動してください。
最初のフレーム表示時にROI選択を行い追跡対象を指定します。
ESCキー押下でプログラム終了、スペースキー押下で追跡対象の再指定を行います。
--use_xxxxを指定することで使用アルゴリズムを追加できます。 ※何も指定していない場合はDaSiamRPNのみで動作

python performance_comparison_sample.py

• --device
  動画ファイル、もしくはカメラデバイス番号の指定
  デフォルト："sample_movie/bird.mp4"
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540
• --use_mil
  MILトラッカーの使用有無
  デフォルト：指定なし
• --use_goturn
  GOTURNトラッカーの使用有無 ※model/GOTURN配下のzip要解凍
  デフォルト：指定なし
• --use_dasiamrpn
  DaSiamRPNトラッカーの使用有無
  デフォルト：指定なし
• --use_csrt
  CSRTトラッカーの使用有無
  デフォルト：指定なし
• --use_kcf
  KCFトラッカーの使用有無
  デフォルト：指定なし
• --use_boosting
  Boostingトラッカーの使用有無
  デフォルト：指定なし
• --use_mosse
  MOSSEトラッカーの使用有無
  デフォルト：指定なし
• --use_medianflow
  MedianFlowトラッカーの使用有無
  デフォルト：指定なし
• --use_tld
  TLDトラッカーの使用有無
  デフォルト：指定なし

ToDo

• GOTURNトラッカーのモデル読み込みに失敗する問題の調査

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

OpenCV-Object-Tracker-Sample is under Apache-2.0 License.

License(Image)

サンプル動画はNHKクリエイティブ・ライブラリーのハクセキレイ　エサをついばみながら歩くを使用しています。