SmrNeuralProvider.h

浏览该文件的文档.

#pragma once
 
#include "animation/AnimSmrNeural.h"
#include "animation/tensor/SmrFeatures.h"
#include "animation/tensor/SmrMeshRetNet.h"
#include "animation/tensor/SmrOnnxRunner.h"
 
#include <string>
#include <vector>
 
namespace eve::animation {
 
class SmrNeuralProvider final : public ISmrNeuralRetarget {
public:
    SmrNeuralProvider() = default;
 
    [[nodiscard]] std::string name() const override;
 
    [[nodiscard]] bool isReady(const AnimRetargetProfile* profile) const override;
 
    [[nodiscard]] Result<SmrNeuralResult> retarget(const SmrNeuralRequest& request) override;
 
private:
    mutable SmrOnnxRunner onnx_;
    mutable std::string   loadedPath_;
    mutable bool          modelLoaded_ = false;
    SmrMeshRetNet         net_{};
 
    [[nodiscard]] Result<void>            ensureOnnx(const AnimRetargetProfile& profile) const;
    [[nodiscard]] Result<SmrNeuralResult> applyRot6d(const SmrNeuralRequest& request, const SmrFeatureBatch& features,
                                                     const std::vector<float>& rot6d, const std::string& backend) const;
};
 
}  // namespace eve::animation