saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-30 03:37:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

patched the logic of removing the ._*.onnx file, in case it comes in position other than the first in listdir (#484)

Browse source
This commit is contained in:
shahasad 2019-02-15 16:08:20 -08:00 committed by GitHub
parent 1f1dcc352f
commit ee702bd288
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
csharp/test/Microsoft.ML.OnnxRuntime.Tests/InferenceTest.cs
View file

@ -237,12 +237,20 @@ namespace Microsoft.ML.OnnxRuntime.Tests
try
{
var onnxModelNames = modelDir.GetFiles("*.onnx");
if (onnxModelNames.Count() != 1)
if (onnxModelNames.Length > 1)
{
// TODO remove file "._resnet34v2.onnx" from test set
if (onnxModelNames[0].Name == "._resnet34v2.onnx")
onnxModelNames[0] = onnxModelNames[1];
else
bool validModelFound = false;
for (int i = 0; i < onnxModelNames.Length; i++)
{
if (onnxModelNames[i].Name != "._resnet34v2.onnx")
{
onnxModelNames[0] = onnxModelNames[i];
validModelFound = true;
}
}
if (!validModelFound)
{
var modelNamesList = string.Join(",", onnxModelNames.Select(x => x.ToString()));
throw new Exception($"Opset {opset}: Model {modelDir}. Can't determine model file name. Found these :{modelNamesList}");