example_person_detect.cpp

Go to the documentation of this file.

#include <sstream>
#include <iomanip>
#include <vector>
#include <algorithm>
 
#include <sfe_toolkit/sfe_core.h>
#include <sfe_toolkit/sfe_person.h>
 
#include "utils.hpp"
#include "solvers.h"
#include "annotate.hpp"
#include "strings.hpp"
 
// Solver parameters, we are using ONNX Runtime CPU backend
const auto SOLVER_PARAMETERS = std::vector<SFESolverParameter>{
    SFESolverParameter { "intra_threads", "4" },
};
 
std::string image_path = "assets/person/person_detect_fisheye.jpg";
std::string solver_detect = SOLVER_PERSON_DETECT;
 
int main() {
    SFEError error;
 
    SFEImage image{};
    DEFER(sfeImageFree(image));
    { // STAGE 1: Load Image
        auto encoded_image = utils::readFile(image_path);
        error = sfeImageDecode(encoded_image.data(), encoded_image.size(), &image);
        utils::checkError(error);
 
        std::cout << "Image size: " << image.width << "x" << image.height << std::endl;
    }
 
    std::vector<SFEDetection> detections(10);
    { // STAGE 2: Run person detection
        // Initialize detection solver
        SFESolver detection_solver{};
        DEFER(sfeSolverFree(detection_solver));
        error = sfeSolverCreate(solver_detect.c_str(), SOLVER_PARAMETERS.data(), SOLVER_PARAMETERS.size(), &detection_solver);
        utils::checkError(error);
 
        // Run detection for up to 10 persons
        size_t detection_count = detections.size();
        float detection_threshold = 0.3;
        error = sfeDetect(detection_solver,
            image,
            detection_threshold,
            detections.data(), &detection_count);
        utils::checkError(error);
        detections.resize(detection_count);
 
        std::cout << "Detected " << detection_count << " persons" << std::endl;
    }
}