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

---

mainImSet.cpp

Go to the documentation of this file.

#ifndef REPOSITORY_USED // Here comes the deprecated stuff
#else // REPOSITORY_USED
#define REPOSITORY_TODO
#endif // REPOSITORY_USED

#include "Core/ImageSet/MakeImageSet.h"
#include "Core/ImageSet/ClusterFeatures.h"
#ifndef REPOSITORY_TODO
#include "Core/ImageSet/IndexConcepts.h"
#include "Core/ImageSet/CombineConcepts.h"
#endif
#include "Util/Channel.h"
#include "Link/Mpi/MpiFuncs.h"
#include "Basis/ILog.h"
#include "Core/ImageSet/CheckSizes.h"
#include "Core/ImageSet/FixSizes.h"
#include "Core/ImageSet/IndexFeatures.h"
#include "Core/ImageSet/ProtoSimilarityEval.h"
#include "Core/ImageSet/Archive.h"
#include "Core/ImageSet/Thumbnails.h"
#include "Core/ImageSet/Reporter.h"
#include "Core/ImageSet/Walker.h"
#ifndef REPOSITORY_TODO
#include "Core/ImageSet/ApplyConcepts.h"
#endif
#include "Core/ImageSet/ConcatFeatures.h"
#ifndef REPOSITORY_TODO
#include "Core/ImageSet/ComputeKernelData.h"
#endif
#include "Core/Feature/VisSem.h"
#ifndef REPOSITORY_TODO
#include "Core/ImageSet/InterestPointProc.h"
#include "Core/ImageSet/MakeQuidLookup.h"
#include "Core/Feature/InterestPointFeature.h"
#endif

// since we are not using libraries:
#include "Link/ImpalaLib.cpp"
#ifndef REPOSITORY_TODO
#include "Link/Svm/LinkSvm.cpp"
#endif

namespace Impala
{
namespace Application
{


using namespace Impala::Core::Array;
using namespace Impala::Core::Feature;
using namespace Impala::Core::ImageSet;

int
mainImSet(int argc, char* argv[])
{
    Link::Mpi::Init(&argc, &argv);

    CmdOptions& options = CmdOptions::GetInstance();
    options.Initialise(false, false, true);
    options.AddOption(0, "report", "perNr", "0");
    options.AddOption(0, "startDir", "idx", "0");
    options.AddOption(0, "numberDirs", "nr", "-1");
    options.AddOption(0, "startFile", "idx", "0");
    options.AddOption(0, "numberFiles", "nr", "-1");
    options.AddOption(0, "virtualWalk", "", "0");
    options.AddOption(0, "imArchive", "", "0");
    options.AddOption(0, "imFileArchive", "", "0");
    options.AddOption(0, "imSplitArchive", "", "0");
    options.AddOption(0, "imCacheSize", "size", "128");
    options.AddOption(0, "precomputed", "[applyconcepts only] boolean: if 1 modelfiles must be for precomputed kernel", "0");
    options.AddOption(0, "storeKernelData", "[applyconcepts only] boolean: if 1 kernel data will be written to disk", "0");
    options.AddOption(0, "addToName", "[fixsizes only] string: add this to the standard filename of archive", "");

    // initialization of options
#ifndef REPOSITORY_TODO
    Impala::Core::ImageSet::InterestPointProc::AddCmdOptions();
#endif

    String usageStr = String("cmd imageSet|ixs\n") +
        "cmd == fixsizes --> imset fixsizes imageSet [max-width max-height [split]]\n" +
        "other commands are undocumented..." +
        "\n\n";
    if (! options.ParseArgs(argc, argv, usageStr, 2))
    {
        Link::Mpi::Finalize();
        return 1;
    }

    ILOG_VAR(Impala.Application.ImSet.mainImSet);

    String cmd = options.GetArg(0);
    String fileName = options.GetArg(1);
    ImageSet* imageSet = MakeImageSet(fileName);
    imageSet->SetImageSrc(options.GetBool("imArchive"),
                          options.GetBool("imFileArchive"),
                          options.GetBool("imSplitArchive"));

    Walker walker(imageSet);
    Reporter* reporter = new Reporter(0);
    walker.AddListener(reporter);

    ILOG_INFO("imset command : " << cmd);

#ifndef REPOSITORY_TODO
    if (cmd == "computekerneldata")
        walker.AddListener(new ComputeKernelData(reporter, options, imageSet));
    if (cmd == "applyconcepts")
        walker.AddListener(new ApplyConcepts(reporter, options, imageSet));
    if (cmd == "indexconcepts")
        walker.AddListener(new IndexConcepts(reporter, options));
    if (cmd == "combineconcepts")
        walker.AddListener(new CombineConcepts(reporter, options));
#endif
    if (cmd == "archive")
        walker.AddListener(new Archive(reporter, options));
    if (cmd == "fixsizes")
        walker.AddListener(new FixSizes(reporter, options));
    if (cmd == "checksizes")
        walker.AddListener(new CheckSizes(reporter, options));
    if (cmd == "concatfeatures")
        walker.AddListener(new ConcatFeatures(reporter, options));
    if (cmd == "indexfeatures")
        walker.AddListener(new IndexFeatures(reporter, options));
#ifndef REPOSITORY_TODO
    if (cmd == "interestpointproc")
        walker.AddListener(new InterestPointProc(reporter, options));
    if (cmd == "makequidlookup")
        walker.AddListener(new MakeQuidLookup(reporter, options));
#endif
    if (cmd == "thumbnails")
        walker.AddListener(new Thumbnails(reporter, options));
    if (cmd == "clusterfeatures")
    {
        if (options.GetNrArg() < 4)
            ILOG_ERROR("cluster: missing argument");
        String feature = options.GetArg(2);
        Computor* c = 0;
        if (feature == "weibull")
            c = new VisSem("vissem", options);
        if (feature == "gabor")
            c = new Core::Feature::VisSem("vissemgabor", options);
        String clusteror = options.GetArg(3);
        walker.AddListener(new ClusterFeatures(c, clusteror, reporter, options));
    }

    if (cmd == "viseval")
    {
        Computor* c = new Core::Feature::VisSem("vissem", options);
        walker.AddListener(new ProtoSimilarityEval(c, reporter, options));
    }
    if (cmd == "visgaboreval")
    {
        Computor* c = new Core::Feature::VisSem("vissemgabor", options);
        walker.AddListener(new ProtoSimilarityEval(c, reporter,options));
    }

    if ((walker.NrListeners() == 1) && (cmd != "nop"))
    {
        ILOG_ERROR("Unknow command : " << cmd);
        Link::Mpi::Finalize();
        return 1;
    }

    walker.AddListener(new Reporter(reporter));
    if (FileNameExt(fileName) == "ixs")
    {
        walker.LoadBookmarks(fileName);
        walker.DoWalkBookmarks();
    }
    else
    {
        int startDir = options.GetInt("startDir");
        int numberDirs = options.GetInt("numberDirs");
#ifdef MPI_USED
        int numProcs = Link::Mpi::NrProcs();
        int myId = Link::Mpi::MyId();
        if (numberDirs == -1)
            numberDirs = imageSet->NrDirs();
        if (startDir + numberDirs >= imageSet->NrDirs())
            numberDirs = imageSet->NrDirs() - startDir;
        int lastDir = startDir + numberDirs;
        int taskSize = numberDirs / numProcs;
        int restSize = numberDirs % numProcs;
        int curDir = startDir;
        for (int i=0 ; i<=myId ; i++)
        {
            startDir = curDir;
            numberDirs = taskSize;
            if (--restSize >= 0)
                numberDirs++;
            if (startDir + numberDirs > lastDir)
                numberDirs = lastDir - startDir;
            curDir += numberDirs;
        }
        ILOG_INFO("Process " << myId << " of " << numProcs
                  << " is doing dirs " << startDir << " till "
                  << startDir + numberDirs);
#endif
        walker.DoWalk(startDir, numberDirs);
    }
    reporter->FinalReport();
    int nrOfErrors = ILOG_ERROR_COUNT;

    nrOfErrors = Link::Mpi::ReduceSum(nrOfErrors);
    ILOG_INFO_ONCE("Root: total nr error = " << nrOfErrors);
    Link::Mpi::Finalize();

    return nrOfErrors;
}

} // namespace Application
} // namespace Impala

int
main(int argc, char* argv[])
{

#ifdef NO_DIALOGONCRASH
#ifdef WIN32
    DWORD oldMode = ::SetErrorMode(SEM_NOGPFAULTERRORBOX);
    ::SetErrorMode(oldMode | SEM_NOGPFAULTERRORBOX);
#endif
#endif

    return Impala::Application::mainImSet(argc, argv);
}

---