SmrOnnxRunner.h

浏览该文件的文档.

#pragma once
 
#include "animation/tensor/SmrFeatures.h"
#include "common/Result.h"
#include "tensor/OnnxModel.h"
 
#include <memory>
#include <span>
#include <string>
#include <vector>
 
namespace eve::animation {
 
class SmrOnnxRunner {
public:
    [[nodiscard]] Result<void>               load(std::span<const uint8_t> bytes);
    [[nodiscard]] Result<void>               loadFile(const std::string& path);
    [[nodiscard]] bool                       isLoaded() const { return static_cast<bool>(model_); }
    [[nodiscard]] tensor::OnnxModelInfo      info() const;
    [[nodiscard]] Result<std::vector<float>> run(const SmrFeatureBatch& features) const;
 
private:
    std::unique_ptr<tensor::OnnxModel> model_;
};
 
}  // namespace eve::animation