StrongSort (OSNET) が YOLOv5、v7、v8 と連携してマルチターゲット追跡を実現する詳細説明
- このプロジェクトには、さまざまな展開シナリオに合わせて調整できる、高度に構成可能な 2 段階のトラッカーが含まれています。 YOLOv5、YOLOv7、YOLOv8 (COCO データセットで事前トレーニングされた一連の物体検出アーキテクチャとモデル) によって生成された検出は StrongSORT に渡され、OSNet に基づいて動きと外観の情報を組み合わせて物体を追跡します。 Yolov5 モデルが検出するようにトレーニングされているあらゆるオブジェクトを追跡できます。
- この記事では
yolov7
追跡は実行されていませんが、yolov5、yolov8
追跡機能とカウント機能は直接利用できます - このうち、yolov5の追跡結果を下図に示しますが、その他の結果については記事を参照してください。
Strongnet + yolov5 でマルチターゲット追跡を実現
環境設定
#新建虚拟环境
conda create -n v7_sort python=3.9