From e8d181e0d17eacaadbdd41ec7eb67821f201f923 Mon Sep 17 00:00:00 2001 From: "feihu.wang" Date: Sat, 11 Jan 2020 14:42:22 +0800 Subject: [PATCH] =?UTF-8?q?prefs:=20=E6=9C=BA=E5=99=A8=E5=AD=A6=E4=B9=A0?= =?UTF-8?q?=E5=8A=A0=E8=BD=BD=E8=B7=AF=E5=BE=84=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: feihu.wang --- .../radar/service/impl/dnn/TensorDnnEstimator.java | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/radar-service-impl/src/main/java/com/pgmmers/radar/service/impl/dnn/TensorDnnEstimator.java b/radar-service-impl/src/main/java/com/pgmmers/radar/service/impl/dnn/TensorDnnEstimator.java index 25444e9..29eca6b 100644 --- a/radar-service-impl/src/main/java/com/pgmmers/radar/service/impl/dnn/TensorDnnEstimator.java +++ b/radar-service-impl/src/main/java/com/pgmmers/radar/service/impl/dnn/TensorDnnEstimator.java @@ -7,6 +7,7 @@ import com.pgmmers.radar.vo.model.ModelConfVO; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; import org.tensorflow.SavedModelBundle; import org.tensorflow.Session; @@ -25,6 +26,9 @@ public class TensorDnnEstimator implements Estimator { private ModelConfService modelConfService; private Map modelBundleMap = new HashMap<>(); + @Value("${sys.conf.workdir}") + private String workDir; + @Override public float predict(Long modelId, Map> data) { ModelConfVO mold = modelConfService.getByModelId(modelId); @@ -77,7 +81,9 @@ public class TensorDnnEstimator implements Estimator { private synchronized SavedModelBundle loadAndCacheModel(ModelConfVO mold) { SavedModelBundle modelBundle = modelBundleMap.get(mold.getId()); if (modelBundle == null) { - File file = new File(mold.getPath()); + String path = workDir + "\\" + mold.getPath(); + String decomposePath = path.substring(0, path.lastIndexOf(".")); + File file = new File(decomposePath); if (file.exists() && file.isDirectory()) { // 模型加载，比较耗时 try {