Parcourir la source

model service

丁云鹏 il y a 5 mois
Parent
commit
f39f2b66ee

+ 1 - 1
recommend-model-jni/paddle/build_testJar.sh

@@ -6,6 +6,6 @@ cd ../src/main/java/com/baidu/paddle/inference
 javac Config.java Predictor.java Tensor.java
 cd ../../../../../../../
 cd paddle
-cp bulid/libpaddle_inference.so .
+cp build/libpaddle_inference.so .
 pwd
 jar cvf JavaInference.jar -C ../src/main/java/ .

+ 11 - 10
recommend-model-jni/src/test/java/com/baidu/paddle/inference/test.java

@@ -9,8 +9,9 @@ public class test {
     }
 
     public static void main(String[] args) {
-        Config config = new Config();
 
+        long time1 = System.currentTimeMillis();
+        Config config = new Config();
         config.setCppModel(args[0], args[1]);
         config.enableMemoryOptim(true);
         config.enableProfile();
@@ -40,15 +41,12 @@ public class test {
 
         predictor.tryShrinkMemory();
         predictor.clearIntermediateTensor();
-
-        System.out.println("predictor1: " + outData[0]);
+        long time2 = System.currentTimeMillis();
+        System.out.println("predictor1: " + outData[0] + ",time:" + (time2 - time1));
         System.out.println("predictor1: " + outData.length);
 
         test(predictor);
-
-        outHandle.destroyNativeTensor();
-        inHandle.destroyNativeTensor();
-        predictor.destroyNativePredictor();
+        test(predictor);
 
         Config newConfig = new Config();
         newConfig.setCppModelDir("/model_dir");
@@ -62,8 +60,10 @@ public class test {
 
     }
 
-    private static void test(Predictor predictor) {
-        Predictor predictor2 = Predictor.clonePaddlePredictor(predictor);
+    private static void test(Predictor predictor2) {
+
+        long time1 = System.currentTimeMillis();
+        Predictor predictor = Predictor.clonePaddlePredictor(predictor2);
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
         inHandle.reshape(4, new int[]{1, 3, 224, 224});
@@ -78,7 +78,8 @@ public class test {
         predictor.tryShrinkMemory();
         predictor.clearIntermediateTensor();
 
-        System.out.println("predictor2: " + outData[0]);
+        long time2 = System.currentTimeMillis();
+        System.out.println("predictor2: " + outData[0] + ",time:" + (time2 - time1));
         System.out.println("predictor2: " + outData.length);
         outHandle.destroyNativeTensor();
         inHandle.destroyNativeTensor();