DepthModel.hpp

Go to the documentation of this file.

#pragma once
 
#include "EyeAICore/TensorBuffer.hpp"
#include "EyeAICore/tflite/TfLiteRuntime.hpp"

class DepthModel {
  public:
    [[nodiscard]] static tl::
        expected<std::unique_ptr<DepthModel>, TfLiteCreateRuntimeError>
        create(
            std::vector<int8_t>&& model_data,
            std::string_view gpu_delegate_serialization_dir,
            std::string_view model_token,
            TfLiteLogWarningCallback log_warning_callback,
            TfLiteLogErrorCallback log_error_callback,
            bool enable_npu,
            std::string npu_skel_directory
        );

    DepthModel(std::unique_ptr<TfLiteRuntime>&& runtime)
        : runtime(std::move(runtime)) {}
 
    DepthModel(DepthModel&&) = default;
    DepthModel& operator=(DepthModel&&) = default;
 
    DepthModel(const DepthModel&) = delete;
    DepthModel& operator=(const DepthModel&) = delete;
 
    ~DepthModel() = default;
 
    using RunResult = tl::expected<
        FloatTensorBuffer<FloatTensorFormat::RelativeDepth>,
        TfLiteRunInferenceError>;

    [[nodiscard]] RunResult
    run(FloatTensorBuffer<FloatTensorFormat::ImageRGB255>& input);
 
    using RunRawResult = tl::expected<
        FloatTensorBuffer<FloatTensorFormat::RawRelativeDepth>,
        TfLiteRunInferenceError>;

    [[nodiscard]] RunRawResult
    run_raw(FloatTensorBuffer<FloatTensorFormat::ImageRGB255>& input);
 
    [[nodiscard]] std::span<const int> get_input_shape() const;
 
    [[nodiscard]] std::span<const int> get_output_shape() const;
 
  private:
    std::unique_ptr<TfLiteRuntime> runtime;
};