能否提供一个使用onnx-engine或者pytorch引擎运行onnx模型的案例 #37
Description
paddlepaddle-engine
你好,你提供的asr_sdk使用的是paddlepaddle-engine运行
<!-- PaddlePaddle -->
<dependency>
<groupId>ai.djl.paddlepaddle</groupId>
<artifactId>paddlepaddle-engine</artifactId>
<version>${djl.version}</version>
</dependency>
package me.aias.example.utils;
import ai.djl.Device;
import ai.djl.ndarray.NDArray;
import ai.djl.repository.zoo.Criteria;
import ai.djl.training.util.ProgressBar;
import org.apache.commons.lang3.tuple.Pair;
import java.nio.file.Path;
import java.nio.file.Paths;
/**
*
* @author Calvin
*
* @email 179209347@qq.com
**/
public class SpeechRecognition {
public SpeechRecognition() {}
public Criteria<NDArray, Pair> criteria() {
Criteria<NDArray, Pair> criteria =
Criteria.builder()
.setTypes(NDArray.class, Pair.class)
.optModelPath(Paths.get("models/deep_speech.zip"))
.optTranslator(new AudioTranslator())
.optEngine("PaddlePaddle") // Use PaddlePaddle engine
.optProgress(new ProgressBar())
.build();
return criteria;
}
}
能否提供一个使用onnx-engine或者pytorch引擎运行onnx模型的案例
1.将paddle模型转为onnx模型
2.使用使用onnx-engine或者pytorch引擎运行onnx模型