DUICUO

ターゲット追跡ライブラリ PySOT がオープンソース化され、コードが正式に利用可能になりました。

先日、CVerは「重大ニュース!SenseTimeがPySOTオブジェクトトラッキングライブラリをオープンソース化:SiamRPN++やSiamMaskなどのアルゴリズムを搭載」という記事を公開し、SenseTimeのSTVIR(SenseTime Video Intelligence Researchチーム)が開発したオープンソースのオブジェクトトラッキングライブラリPySOTを紹介しました。当時、PySOTのコードはまだアップロードされていなかったため、前回の記事では新機能の紹介のみでしたが、それでも大きな注目を集めました。

2019年5月末にコードが公開されるだろうと思われていましたが、2019年5月13日の早朝にPySOTが正式にコードを公開しました!

https://github.com/STVIR/pysot

Amsi はこれまでにもターゲット追跡に関する優れた記事をいくつか公開してきましたが、この記事では誰もが興味を持つであろう、紙とコードのプロジェクトを紹介します。

  • シャムネットワークに基づくターゲット追跡アルゴリズムの概要
  • GitHub: ターゲット追跡に関する最も包括的なリソースコレクション
  • CVPR 2019 | Siamese Networks (SiamMask) についての考察

PySOT の機能

PySOTオブジェクトトラッキングライブラリは、複数の最先端(SOTA)単一オブジェクトトラッキングアルゴリズムを実装しています。このプロジェクトはPythonで記述されており、PyTorchディープラーニングフレームワークをベースにしています。

このプロジェクトには、トラッカーを評価するための Python ツール インターフェースも含まれています。

なぜPySOTという名前なのでしょう?Python/PyTorchとSingle Object Trackingを組み合わせたものなので、おそらくご想像の通りでしょう。AmusiはPyMOTを将来オープンソース化することを強く推奨しています(笑)。

PySOT には 5 つのアルゴリズムが含まれています。

  • サイアムマスク(CVPR 2019)
  • SiamRPN++ (CVPR 2019 口頭発表)
  • DaSiamRPN (ECCV 2018)
  • サイアムRPN (CVPR 2018)
  • サイアムFC (ECCV 2016)

サイアムRPN++

PySOT は 3 つのバックボーンを提供します。

  • ResNet{18, 34, 50}
  • モバイルネットV2
  • アレックスネット

PySOT の評価ツールは次のデータセットをサポートしています。

  • OTB2015
  • 投票16/18/19
  • VOT18-LT
  • ラソット
  • 無人航空機123

PySOTのインストールとテスト

更新された PySOT には、インストール手順、モデル ダウンロード ライブラリ、テスト/評価スクリプトなどが含まれています。

PySOT のインストールと環境の依存関係

  • Python 3.7 を使用した Conda。
  • Nvidia GPU。
  • パイトーチ 0.4.1
  • ヤックス
  • pyyamlmatplotlib
  • tqdm
  • オープンCV
  1. # 環境を作成してアクティブ化conda create --name pysot python= 3 .7conda activate pysot# numpy/pytorch/opencvcondaをインストール install numpyconda install pytorch= 0.4 . 1 torchvision cuda90 -c pytorchpip install opencv-python# その他の要件をインストールpip install pyyaml ​​yacs tqdm colorama matplotlib cython tensorboardX# 拡張機能をビルドpython setup.py build_ext --inplace# スクリプトで試すbash install.sh /path/to/your/conda pysot

PySOTモデル動物園

視覚追跡ベースライン(短期追跡)

PySOT ツールセット: 評価

例: VOT2018-LT

テストスクリプト

  1. cd experiments/siamrpn_r50_l234_dwxcorrpython -u ../../tools/test.py --snapshot model.pth # モデルパス --dataset VOT2018 # データセット名 --config config.yaml # 設定ファイル

カンカン

PySOT のコードが公開されました。ぜひテストしてみてください。