example_tattoo_detect.cpp File Reference

#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"

Go to the source code of this file.

Functions
int	main ()

Variables
const auto	SOLVER_PARAMETERS
std::string	image_path = "assets/tattoo/tattoo_0.png"
std::optional< std::string >	solver = std::nullopt

Function Documentation

main()

int main

(

)

[Tattoo Detect]

[Tattoo Detect]

Definition at line 26 of file example_tattoo_detect.cpp.

           {
  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;
  }
}

Variable Documentation

image_path

std::string image_path = "assets/tattoo/tattoo_0.png"

Definition at line 19 of file example_tattoo_detect.cpp.

solver

std::optional<std::string> solver = std::nullopt

Definition at line 23 of file example_tattoo_detect.cpp.

SOLVER_PARAMETERS

const auto SOLVER_PARAMETERS

Initial value:

= std::vector<SFESolverParameter>{
    SFESolverParameter{"intra_threads", "4"},
}

Definition at line 15 of file example_tattoo_detect.cpp.