Struct OnnxEngine 

pub struct OnnxEngine { /* private fields */ }

ONNX 推論エンジン

Implementations

impl OnnxEngine

pub fn load( model_path: &Path, config: &VoiceConfig, device: &str, ) -> Result<Self, PiperError>

ONNX モデルを読み込んでエンジンを初期化する。

device は "cpu", "auto", "cuda", "cuda:0", "coreml", "directml", "tensorrt" のいずれか。 "auto" 指定時は CUDA を試行し、失敗すれば CPU にフォールバックする。

pub fn capabilities(&self) -> &ModelCapabilities

モデルの能力情報を返す

pub fn sample_rate(&self) -> u32

サンプルレートを返す

pub fn synthesize( &mut self, request: &SynthesisRequest, ) -> Result<SynthesisResult, PiperError>

ONNX 推論を実行して音声を生成する。

ONNX 入力テンソル順序:

1. input (phoneme_ids): int64 [1, phoneme_length]
2. input_lengths: int64 [1]
3. scales: float32 [3] = [noise_scale, length_scale, noise_w]
4. sid (条件付き): int64 [1] – has_sid が true のとき
5. lid (条件付き): int64 [1] – has_lid が true のとき
6. prosody_features (条件付き): int64 [1, phoneme_length, 3]

ONNX 出力:

• output: float32 [1, 1, audio_samples]
• durations (オプション): float32 [1, phoneme_length]