File tree 1 file changed +3
-0
lines changed
src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020
1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change 70
70
import org .tensorflow .proto .framework .MetaGraphDef ;
71
71
import org .tensorflow .proto .framework .SignatureDef ;
72
72
import org .tensorflow .proto .framework .TensorInfo ;
73
+ import org .yaml .snakeyaml .Yaml ;
73
74
74
75
/**
75
76
* Class to that communicates with the dl-model runner, see
@@ -653,13 +654,15 @@ private static String getCurrentClasspath() throws UnsupportedEncodingException
653
654
String gsonPath = getPathFromClass (Gson .class );
654
655
String jnaPath = getPathFromClass (com .sun .jna .Library .class );
655
656
String jnaPlatformPath = getPathFromClass (com .sun .jna .platform .FileUtils .class );
657
+ String snakeYaml = getPathFromClass (Yaml .class );
656
658
if (modelrunnerPath == null || (modelrunnerPath .endsWith ("DeepLearningEngineInterface.class" )
657
659
&& !modelrunnerPath .contains (File .pathSeparator )))
658
660
modelrunnerPath = System .getProperty ("java.class.path" );
659
661
String classpath = modelrunnerPath + File .pathSeparator + imglib2Path + File .pathSeparator ;
660
662
classpath = classpath + gsonPath + File .pathSeparator ;
661
663
classpath = classpath + jnaPath + File .pathSeparator ;
662
664
classpath = classpath + jnaPlatformPath + File .pathSeparator ;
665
+ classpath = classpath + snakeYaml + File .pathSeparator ;
663
666
664
667
return classpath ;
665
668
}
You can’t perform that action at this time.
0 commit comments