p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48226 [MS][LITE][parallel predict] Model API: Add java ST 

Merge pull request !48226 from yefeng/512-Add_java_model_ST
This commit is contained in:
i-robot 2023-01-31 06:11:53 +00:00 committed by Gitee
parent 5cc5607898 390d6e06bb
commit 07992a01b2
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/test/config_level0/models_java_cpu_cloud.cfg
View File

@ -1 +1,2 @@
yolox.onnx;1:input;1,3,640,640;; yolox.onnx;1:input;1,3,640,640;;
screenshot_text_location_220613_100.onnx;1:input_images;1,3,1024,1024;;

1
mindspore/lite/test/config_level0/models_python_cpu.cfg
View File

@ -1 +1,2 @@
yolox.onnx;1:input;1,3,640,640;; yolox.onnx;1:input;1,3,640,640;;
screenshot_text_location_220613_100.onnx;1:input_images;1,3,1024,1024;;

3
mindspore/lite/test/st/java/src/main/java/Benchmark.java
View File

@ -160,7 +160,6 @@ public class Benchmark {
runner.free(); runner.free();
return; return;
} }
System.out.println("========== model parallel runner predict success ==========");
config.free(); config.free();
for (int i = 0; i < inputs.size(); i++) { for (int i = 0; i < inputs.size(); i++) {
inputs.get(i).free(); inputs.get(i).free();
@ -169,7 +168,7 @@ public class Benchmark {
outputs.get(i).free(); outputs.get(i).free();
} }
runner.free(); runner.free();
return; System.out.println("========== model parallel runner predict success ==========");
} }
MSContext context = new MSContext(); MSContext context = new MSContext();