leng-yue/ONNX-TensorRT-DLL

请问这里输入输出都要转为一维数组吗

shzql opened this issue · 1 comments

shzql commented
Mat source = new Mat(@"./data/test.jpg", ImreadModes.Color);
            source = source.Resize(new OpenCvSharp.Size(128, 128));
            float[] input = Image.Normalize(source);
            float[] output = new float[1];

嗯, 毕竟 TensorRT 的输入输出都是一维数组