inference_core.h

Go to the documentation of this file.

 
#ifndef SRC_INFERENCE_CORE_H_
#define SRC_INFERENCE_CORE_H_
 
#include <interpreter_builder.h>
#include <stdint.h>
 
#include <memory>
 
#include "intermediate_structures.h"
#include "pre_processor.h"
 
namespace Inference {
 
class InferenceCore {
 private:
  size_t model_input_width;
  size_t model_input_height;
  size_t model_input_size;
  uint8_t model_input_channels = 3;
  std::unique_ptr<tflite::Interpreter> interpreter;
  tflite::StderrReporter error_reporter;
  std::unique_ptr<tflite::FlatBufferModel> model;
 
  Coordinate pixel_coord_to_Coordinate(uint32_t x, uint32_t y,
                                       float confidence);
 
 public:
  InferenceCore(const char* model_filename, size_t model_input_width,
                size_t model_input_height);
 
  InferenceResults run(PreProcessing::PreProcessedImage preprocessed_image);
};
}  // namespace Inference
 
#endif  // SRC_INFERENCE_CORE_H_