testMatcher.cpp

#include <iostream>
 
#include "exception.h"
#include "decoder.h"
#include "detector.h"
#include "extractor.h"
#include "matcher.h"
 
 
int testMatcher_1x1(const std::string& path0, const std::string& path1)
{
    try {
        auto cfg = Diametrix::makeConfig("test.cfg");
        auto decoder = Diametrix::createDecoder(cfg);
        auto detector = Diametrix::createDetector(cfg);
        auto extractor = Diametrix::createExtractor(cfg);
        auto matcher = Diametrix::createMatcher(cfg);
 
        std::shared_ptr<Diametrix::Template> template0;
        {
            std::cout << path0.c_str() << std::endl;
 
            auto image = decoder->read(path0);
            if(image->empty()) {
                std::cout << "The image is empty" << std::endl;
                return -3;
            }
                      
            std::vector<std::shared_ptr<Diametrix::Detection>> detections;
            detector->detect(image, 500, 0.1, detections);
 
            if(detections.size()) {
                extractor->extract(detections[0], template0);
            } else {
                Diametrix::Point fingerprintCenter;
                fingerprintCenter.x = image->width() / 2;
                fingerprintCenter.y = image->height() / 2;
                auto detection = Diametrix::constructDetection(image, 500, fingerprintCenter, 0);
 
                extractor->extract(detections[0], template0);
            }
        }
 
        std::shared_ptr<Diametrix::Template> template1;
        {
            std::cout << path1.c_str() << std::endl;
 
            auto image = decoder->read(path1);
            if(image->empty()) {
                std::cout << "The image is empty" << std::endl;
                return -3;
            }
                      
            std::vector<std::shared_ptr<Diametrix::Detection>> detections;
            detector->detect(image, 500, 0.1, detections);
 
            if(detections.size()) {
                extractor->extract(detections[0], template1);
            } else {
                Diametrix::Point fingerprintCenter;
                fingerprintCenter.x = image->width() / 2;
                fingerprintCenter.y = image->height() / 2;
                auto detection = Diametrix::constructDetection(image, 500, fingerprintCenter, 0);
 
                extractor->extract(detections[0], template1);
            }
        }
 
        std::cout << "Similarity score: " << matcher->match(template0, template1) << std::endl;
 
        return 0;
    }
    catch (const Diametrix::Exception& exp) {
        std::cout << "Exception:\t" << exp.what() << std::endl;
        return -2;
    }
    catch (...) {
        std::cout << "Something happened" << std::endl;
        return -1;
    }
 
    return 0;
}