mymagicpower / AIAS

AIAS (AI Acceleration Suite) - 人工智能加速器套件。提供: 包括SDK,平台引擎,场景套件在内,合计超过100个项目组成的项目集。
Apache License 2.0
649 stars 233 forks source link

能否提供一个使用onnx-engine或者pytorch引擎运行onnx模型的案例 #37

Open litongjava opened 6 months ago

litongjava commented 6 months ago

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模型

mymagicpower commented 6 months ago

https://github.com/mymagicpower/AIAS/blob/main/1_image_sdks/ocr_v4_sdk/src/main/java/top/aias/ocr/utils/detection/OcrV4Detection.java

Original Email

Sender:"Tong Li"< @.*** >;

Sent Time:2023/12/4 16:35

To:"mymagicpower/AIAS"< @.*** >;

Cc recipient:"Subscribed"< @.*** >;

Subject:[mymagicpower/AIAS] 能否提供一个使用onnx-engine或者pytorch引擎运行onnx模型的案例 (Issue #37)

paddlepaddle-engine 你好,你提供的asr_sdk使用的是paddlepaddle-engine运行