testDetector.cpp

#include <iostream>
 
#include "exception.h"
#include "decoder.h"
#include "detector.h"
 
int testDetector(const std::string& path)
{
    try {
        auto cfg = Diametrix::makeConfig("test.cfg");
        auto decoder = Diametrix::createDecoder(cfg);
        auto detector = Diametrix::createDetector(cfg);
 
        auto image = decoder->read(path);
        std::cout << "Image height: " << image->height() << std::endl;
        std::cout << "Image width: " << image->width() << std::endl;
        
        std::vector<std::shared_ptr<Diametrix::Detection>> detections;
        detector->detect(image, 500, 0.1, detections);
 
        std::cout << "Detections:" << std::endl;
        for(unsigned k = 0; k < detections.size(); ++k) {
            std::cout << k << ":\t" << detections[k]->confidence() << "\t" << detections[k]->bbox().x0 << "\t" << detections[k]->bbox().y0
                        << "\t" << detections[k]->bbox().x1 << "\t" << detections[k]->bbox().y1 << 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;
}