C# Onnx파일 임포트 시 오류

Question

using System; using System.Collections.Generic; using System.Linq; using Microsoft.ML; using Microsoft.ML.Data; using Microsoft.ML.Transforms.Onnx; namespace ConsoleApp1 { internal class Program { static string ONNX_MODEL_PATH = @"C:\Users\User\Desktop\ConsoleApp1\ConsoleApp1\seongModel.onnx"; static void Main(string[] args) { MLContext mlContext = new MLContext(); var onnxPredictionPipeline = GetPredictionPipeline(mlContext); var onnxPredictionEngine = mlContext.Model.CreatePredictionEngine, OnnxOutput>(onnxPredictionPipeline); List input = new List(); input.Add(new OnnxInput() { obs_0 = new[] { 1.2f, 124f, 1.0f, 1.1f, 16.2f, 13f, 0.24f, 0.16f } }); var prediction = onnxPredictionEngine.Predict(input); Console.WriteLine($"Predicted Fare: {prediction.continuous_actions.First()}"); } static ITransformer GetPredictionPipeline(MLContext mlContext) { string[] inputColumns = { "obs_0" }; var outputColumns = new string[] { "continuous_actions" }; var onnxPredictionPipeline = mlContext .Transforms .ApplyOnnxModel( outputColumnNames: outputColumns, inputColumnNames: inputColumns, ONNX_MODEL_PATH); var emptyDv = mlContext.Data.LoadFromEnumerable(new List()); return onnxPredictionPipeline.Fit(emptyDv); } public class OnnxInput { [VectorType(8)] [ColumnName("obs_0")] public float[] obs_0 { get; set; } } public class OnnxOutput { [VectorType(2)] [ColumnName("continuous_actions")] public float[] continuous_actions { get; set; } } } }C# .Net환경에서 onnx파일을 가져와 사용하고자 합니다. 하지만, ms에서 알려준대로 따라가봤는데, 자꾸 입력 데이터 이름을 찾을 수 없다해서 막막한 상황입니다. onnx파일 경로는 @"C:\Users\User\Desktop\ConsoleApp1\ConsoleApp1\seongModel.onnx"에 존재하고 디렉토리에 복사를 해놓은 상태입니다. 형님들 도와주십쇼.... 아래는 코드랑 onnx netron에서 확인한 모양이고, 잘 아시는분 도와주십쇼...Netron 형상파일 위치

C# Onnx파일 임포트 시 오류

0개의 답변

#pytorch 목록