MeshRet-compatible ONNX runner on CPU via eve::tensor::OnnxModel. 更多...
#include <SmrOnnxRunner.h>
Public 成员函数 | |
| Result< void > | load (std::span< const uint8_t > bytes) |
| Loads . | |
| Result< void > | loadFile (const std::string &path) |
| Loads file. | |
| bool | isLoaded () const |
| True when loaded. | |
| tensor::OnnxModelInfo | info () const |
| Info. | |
| Result< std::vector< float > > | run (const SmrFeatureBatch &features) const |
| Run. | |
详细描述
MeshRet-compatible ONNX runner on CPU via eve::tensor::OnnxModel.
I/O contract from scripts/export_meshret_onnx.py:
- inputs: source_rot6d [1,T,J,6], source_geom [1,S,7], target_geom [1,S,7], source_dmi [1,T,P,10]
- output: target_rot6d [1,T,J,6]
@ownership Owns the loaded OnnxModel until load/reload or destruction. @lifetime Model bytes are copied at load; feature tensors are borrowed only for run(). @thread Owner thread; not re-entrant.
在文件 SmrOnnxRunner.h 第 25 行定义.
成员函数说明
◆ info()
| tensor::OnnxModelInfo eve::animation::SmrOnnxRunner::info | ( | ) | const |
Info.
在文件 SmrOnnxRunner.cpp 第 49 行定义.
◆ isLoaded()
|
inline |
True when loaded.
在文件 SmrOnnxRunner.h 第 32 行定义.
◆ load()
| Result< void > eve::animation::SmrOnnxRunner::load | ( | std::span< const uint8_t > | bytes | ) |
Loads .
在文件 SmrOnnxRunner.cpp 第 24 行定义.
引用了 bytes, eve::Result< T >::failure(), eve::tensor::OnnxModel::load() , 以及 eve::Result< T >::success().
被这些函数引用 loadFile().
◆ loadFile()
| Result< void > eve::animation::SmrOnnxRunner::loadFile | ( | const std::string & | path | ) |
Loads file.
在文件 SmrOnnxRunner.cpp 第 31 行定义.
引用了 bytes, eve::Diagnostic::error(), eve::Failed, eve::Result< T >::failure(), file, eve::InvalidArgument, load(), eve::NotFound, path , 以及 size.
◆ run()
| Result< std::vector< float > > eve::animation::SmrOnnxRunner::run | ( | const SmrFeatureBatch & | features | ) | const |
Run.
在文件 SmrOnnxRunner.cpp 第 54 行定义.
引用了 bytes, eve::Diagnostic::error(), eve::Failed, eve::animation::SmrFeatureBatch::frames, eve::InvalidArgument, eve::animation::SmrFeatureBatch::joints, eve::animation::SmrFeatureBatch::pairs, eve::PreconditionViolation, eve::animation::SmrFeatureBatch::sensors, eve::animation::SmrFeatureBatch::sourceDmi, eve::animation::SmrFeatureBatch::sourceGeom, eve::animation::SmrFeatureBatch::sourceRot6d , 以及 eve::animation::SmrFeatureBatch::targetGeom.
该类的文档由以下文件生成:
- src/modules/animation/tensor/SmrOnnxRunner.h
- src/modules/animation/tensor/SmrOnnxRunner.cpp