example_tattoo_detect.cpp

Go to the documentation of this file.

#include <algorithm>
#include <iomanip>
#include <sstream>
#include <vector>
 
#include <sfe_toolkit/sfe_core.h>
#include <sfe_toolkit/sfe_tattoo.h>
 
#include "annotate.hpp"
#include "solvers.h"
#include "utils.hpp"
#include "optional"
 
// 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/tattoo/tattoo_0.png";
#ifdef SOLVER_TATTOO_DETECT
std::optional<std::string> solver = SOLVER_TATTOO_DETECT;
#else
std::optional<std::string> solver = std::nullopt;
#endif // SOLVER_TATTOO_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.value_or("").c_str(), SOLVER_PARAMETERS.data(), SOLVER_PARAMETERS.size(),
        &detection_solver);
    utils::checkError(error);
 
    // Run detection for up to 10 tattoos
    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 << " tattoos" << std::endl;
  }
}