dnn test

recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DNNModel.java

@@ -27,8 +27,8 @@ public class DNNModel implements Model {
         String inNames = predictor.getInputNameById(0);
         Tensor inHandle = predictor.getInputHandle(inNames);
         log.info("predictor2 inNames={}", inNames);
-        inHandle.reshape(4, new int[]{1, 3, 224, 224});
-        float[] inData = new float[39];
+        inHandle.reshape(2, new int[]{1, 1});
+        float[] inData = new float[1 * 1];
         inHandle.copyFromCpu(inData);
         predictor.run();