Skip to content

Commit

Permalink
correct big bug creating java path
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosuc3m committed Sep 6, 2023
1 parent 15fde4b commit 46bc89f
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -438,7 +438,7 @@ public static String getModelOutputName(String outputName) {
* @throws RunModelException if there is any error running the model
*/
public static void main(String[] args) throws LoadModelException, IOException, RunModelException {
// Unpack the args needed
// Unpack the args needed
if (args.length < 4)
throw new IllegalArgumentException("Error exectuting Tensorflow 2, "
+ "at least 5 arguments are required:" + System.lineSeparator()
Expand Down Expand Up @@ -597,6 +597,8 @@ private void retrieveInterprocessingTensors(List<Tensor<?>> tensors) throws RunM
private List<String> getProcessCommandsWithoutArgs() throws IOException, URISyntaxException {
String javaHome = System.getProperty("java.home");
String javaBin = javaHome + File.separator + "bin" + File.separator + "java";
if (javaBin.contains(" "))
javaBin = "\"" + javaBin + "\"";

String modelrunnerPath = getPathFromClass(DeepLearningEngineInterface.class);
String imglib2Path = getPathFromClass(NativeType.class);
Expand Down

0 comments on commit 46bc89f

Please sign in to comment.