MultiTrackerの使用

目的

このチュートリアルでは、次のことを学ぶ

• MultiTrackerオブジェクトを作成する。
• MultiTrackerオブジェクトを使って複数のオブジェクトを一度に追跡する。

ソースコード

/*----------------------------------------------
 * Usage:
 * example_tracking_multitracker <video_name> [algorithm]
 *
 * example:
 * example_tracking_multitracker Bolt/img/%04d.jpg
 * example_tracking_multitracker faceocc2.webm KCF
 *--------------------------------------------------*/
 
#include <opencv2/core/utility.hpp>
#include <opencv2/tracking.hpp>
#include <opencv2/videoio.hpp>
#include <opencv2/highgui.hpp>
#include <iostream>
#include <cstring>
#include <ctime>
#include "samples_utility.hpp"
 
using namespace std;
using namespace cv;
 
int main( int argc, char** argv ){
 // show help
 if(argc<2){
    cout<<
 " Usage: example_tracking_multitracker <video_name> [algorithm]\n"
 " examples:\n"
 " example_tracking_multitracker Bolt/img/%04d.jpg\n"
 " example_tracking_multitracker faceocc2.webm MEDIANFLOW\n"
      << endl;
 return 0;
  }
 
 // set the default tracking algorithm
  std::string trackingAlg = "KCF";
 
 // set the tracking algorithm from parameter
 if(argc>2)
    trackingAlg = argv[2];
 
 // create the tracker
 legacy::MultiTracker trackers;
 
 // container of the tracked objects
  vector<Rect2d> objects;
 
 // set input video
  std::string video = argv[1];
 VideoCapture cap(video);
 
 Mat frame;
 
 // get bounding box
  cap >> frame;
  vector<Rect> ROIs;
  selectROIs("tracker",frame,ROIs);
 
 //quit when the tracked object(s) is not provided
 if(ROIs.size()<1)
 return 0;
 
 // initialize the tracker
  std::vector<Ptr<legacy::Tracker> > algorithms;
 for (size_t i = 0; i < ROIs.size(); i++)
  {
      algorithms.push_back(createTrackerByName_legacy(trackingAlg));
      objects.push_back(ROIs[i]);
    }
 
  trackers.add(algorithms,frame,objects);
 
 // do the tracking
  printf("Start the tracking process, press ESC to quit.\n");
 for ( ;; ){
 // get frame from the video
    cap >> frame;
 
 // stop the program if no more images
 if(frame.rows==0 || frame.cols==0)
 break;
 
 //update the tracking result
    trackers.update(frame);
 
 // draw the tracked object
 for(unsigned i=0;i<trackers.getObjects().size();i++)
      rectangle( frame, trackers.getObjects()[i], Scalar( 255, 0, 0 ), 2, 1 );
 
 // show image with the tracked object
    imshow("tracker",frame);
 
 //quit on ESC button
 if(waitKey(1)==27)break;
  }
 
}

説明

1. MultiTrackerオブジェクトの作成

    legacy::MultiTracker trackers;

   スニペットに示すように、MultiTrackerオブジェクトを作成し、追跡するすべてのオブジェクトに対して同じ追跡アルゴリズムを使用できる。各追跡オブジェクトごとに異なる種類の追跡アルゴリズムを使いたい場合は、MultiTrackerオブジェクトに新しいオブジェクトを追加するたびに追跡アルゴリズムを定義する必要がある。

2. 複数オブジェクトの選択

     vector<Rect> ROIs;
    selectROIs("tracker",frame,ROIs);

   コードに示すように、selectROIを使って複数のオブジェクトを選択でき、その結果はcv::Rect2dのベクトルに格納される。

3. 追跡オブジェクトのMultiTrackerへの追加

     std::vector<Ptr<legacy::Tracker> > algorithms;
    for (size_t i = 0; i < ROIs.size(); i++)
     {
         algorithms.push_back(createTrackerByName_legacy(trackingAlg));
         objects.push_back(ROIs[i]);
       }
    
     trackers.add(algorithms,frame,objects);

   コードに示すように、すべての追跡オブジェクトを一度にMultiTrackerに追加できる。この場合、MultiTrackerオブジェクトの宣言で指定された同じ追跡アルゴリズムを使ってすべてのオブジェクトが追跡される。各追跡オブジェクトに異なるトラッカーアルゴリズムを使いたい場合は、追跡オブジェクトを1つずつ追加し、cv::legacy::MultiTracker::addのバリアントを使ってそれぞれの追跡アルゴリズムを指定する必要がある。

   参照

   cv::legacy::MultiTracker::add( const String& trackerType, const Mat& image, const Rect2d& boundingBox )

4. 結果の取得

    // draw the tracked object
    for(unsigned i=0;i<trackers.getObjects().size();i++)
    rectangle( frame, trackers.getObjects()[i], Scalar( 255, 0, 0 ), 2, 1 );

   コードに示すように、MultiTrackerクラスが提供する公開変数cv::legacy::MultiTracker::objectsから結果にアクセスできる。