|| || || || || || || ||

---

mainTrack.cpp

Go to the documentation of this file.

#include <vector>
#include "Basis/CmdOptions.h"
#include "Core/Stream/RgbDataSrcFactory.h"
#include "Core/Array/WritePng.h"

#include "Core/Tracking/TrackerFactory.h"
#include "Core/Tracking/FrameRegionTable.h"
#include "Core/Tracking/Classifier.h"

using namespace Impala;
using namespace Core;
using namespace Util;
using namespace Stream;
using namespace Tracking;

CmdOptions& SetUpOptions(int argc, char *argv[])
{
    CmdOptions& options = CmdOptions::GetInstance();
    
    options.AddOption(0, "tracker", "type", "fore_back", "fore_back|appearance_kalman");
    options.AddOption(0, "tracker-dump-im", "filename", "");

    options.AddOption(0, "fore_back.sampleSpacing", "int", "5");
    options.AddOption(0, "fore_back.filterSize", "int", "19");
    options.AddOption(0, "fore_back.gabor.scale", "int", "4");
    options.AddOption(0, "fore_back.gabor.frequency", "int", "2");
    options.AddOption(0, "fore_back.gabor.directionCount", "int", "2");
    options.AddOption(0, "fore_back.classifier.gamma", "double", "0.05");
    options.AddOption(0, "fore_back.classifier.lambdafactor", "double", "0.004");
    options.AddOption(0, "fore_back.tracker-debug-file", "filename", "debug.txt");

    options.AddOption(0, "appearance_kalman.kalman.chi^2", "double", "13"); //is (Chi_d,delta)^2 where d=3 and delta=0.99
    options.AddOption(0, "appearance_kalman.kalman.gamma", "double", "0.3");
    options.AddOption(0, "appearance_kalman.kalman.N_omax", "int", "40");
    options.AddOption(0, "appearance_kalman.kalman.K", "int", "20");

    options.ParseArgs(argc, argv, "vindeoname inputfile outputfile", 3);
    return options;
}

int main(int argc, char* argv[])
{
    CmdOptions& options = SetUpOptions(argc, argv);
    ILOG_VAR(main);
    std::string vidname = options.GetArg(0);
    std::string input = options.GetArg(1);
    std::string output = options.GetArg(2);
    ILOG_INFO(vidname);
    String imageDumpFilename = options.GetString("tracker-dump-im");
    bool doDump = false;
    if(imageDumpFilename != "")
        doDump = true;

    RgbDataSrcFactory& factory = RgbDataSrcFactory::Instance();
    RgbDataSrc* src = factory.Construct(RgbDataSrcFactory::SRC_LAVC, vidname);
    if (! src->Valid())
    {
        ILOG_ERROR("RgbDataSrc failed");
        return 0;
    }
    ILOG_INFO("opened movie: << " << src->LastFrame()+1 << " frames of " << src->FrameWidth()  << " x " << src->FrameHeight());

    // read frame regions from file.
    FrameRegionTable* regions = new FrameRegionTable;
    std::ifstream ifs(input.c_str());
    if(ifs.is_open())
    {
        regions->ReadFrom(ifs);
        ifs.close();
    }

    Array::Array2dVec3UInt8* image=0;

    for(int i=0 ; i<regions->Size() ; i++)
    {
        FrameRegion region = regions->GetRegion(i);
        ILOG_INFO("starting tracker for region " << i);
        int startFrame = region.GetFrameNumber();
        src->GotoFrame(startFrame);

        //setup tracker
        FrameRegionTable* track = new FrameRegionTable;
        Tracker* tracker = TrackerFactory::CreateTracker(options);
        image = Array::MakeFromData<Array::Array2dVec3UInt8>(src->DataPtr(), src->FrameWidth(), src->FrameHeight());
        tracker->Initialize(image, &region, track);

        //loop tracker until end
        while(!src->TheEnd() && tracker->ObjectVisible())
        {
            ILOG_DEBUG(".");
            delete image;
            image = Array::MakeFromData<Array::Array2dVec3UInt8>(src->DataPtr(), src->FrameWidth(), src->FrameHeight());
            tracker->Process(image);
            if(doDump)
            {
                static int frame = 0;
                WritePng(image, imageDumpFilename + MakeString(frame) + ".png",
                    &Util::Database::GetInstance());
                frame++;
            }
            src->NextFrame();
        }

        delete tracker;
        delete image;
        ILOG_INFO("done tracking, saving data...");

        //save the track
        std::ostringstream oss;
        track->WriteTo(oss);
        Util::Database& db = Util::Database::GetInstance();
        Util::IOBuffer* buf = db.GetIOBuffer(output + MakeString(i) + ".txt", false, false, "tmp");
        buf->Puts(oss.str());
        delete buf;

        track->WriteTo(std::cout);
        delete track;
    }
    ILOG_INFO("all frame regions tracked succesfully");
    return 0;
}

---