From dac5c3dd561ac3fa05fec236e49f443e74349e20 Mon Sep 17 00:00:00 2001
From: skyyan <skyyan2000@163.com>
Date: Fri, 19 Jul 2024 10:59:21 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=20Caused=20by:=20java.net.UR?=
 =?UTF-8?q?ISyntaxException:=20Illegal=20character=20in=20opaque=20part=20?=
 =?UTF-8?q?at=20index=202:?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../translator/AbstractDjlTranslator.java     | 14 +++++++++--
 .../javavision/util/DjlHandlerUtil.java       | 23 +++++++++++++++++++
 .../javavision/util/PathConstants.java        |  2 +-
 3 files changed, 36 insertions(+), 3 deletions(-)
 create mode 100644 src/main/java/com/github/javpower/javavision/util/DjlHandlerUtil.java

diff --git a/src/main/java/com/github/javpower/javavision/detect/translator/AbstractDjlTranslator.java b/src/main/java/com/github/javpower/javavision/detect/translator/AbstractDjlTranslator.java
index 7ba00c2..853c76b 100644
--- a/src/main/java/com/github/javpower/javavision/detect/translator/AbstractDjlTranslator.java
+++ b/src/main/java/com/github/javpower/javavision/detect/translator/AbstractDjlTranslator.java
@@ -4,13 +4,18 @@ import ai.djl.modality.cv.Image;
 import ai.djl.repository.zoo.Criteria;
 import ai.djl.training.util.ProgressBar;
 import ai.djl.translate.Translator;
+import com.github.javpower.javavision.util.DjlHandlerUtil;
 import com.github.javpower.javavision.util.JarFileUtils;
 import com.github.javpower.javavision.util.PathConstants;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 
 
+import java.io.File;
 import java.io.IOException;
 import java.util.Map;
 
+@Slf4j
 public abstract class AbstractDjlTranslator<T> {
 
     public String modelName;
@@ -35,11 +40,15 @@ public abstract class AbstractDjlTranslator<T> {
         } catch (IOException e) {
             throw new RuntimeException(e);
         }
-        String model_path = PathConstants.TEMP_DIR + PathConstants.ONNX + "/" + modelName;
+//        String model_path = PathConstants.TEMP_DIR + PathConstants.ONNX + "/" + modelName;
+        String modelPath = PathConstants.TEMP_DIR + File.separator+PathConstants.ONNX_NAME+ File.separator + modelName;
+        log.info("路径修改前:{}",modelPath);
+        modelPath= DjlHandlerUtil.getFixModelPath(modelPath);
+        log.info("路径修改后:{}",modelPath);
         Criteria<Image, T> criteria =
                 Criteria.builder()
                         .setTypes(Image.class, getClassOfT())
-                        .optModelUrls(model_path)
+                        .optModelUrls(modelPath)
                         .optTranslator(translator)
                         .optEngine(getEngine()) // Use PyTorch engine
                         .optProgress(new ProgressBar())
@@ -53,5 +62,6 @@ public abstract class AbstractDjlTranslator<T> {
     protected abstract Class<T> getClassOfT();
 
     protected abstract String getEngine();
+
 }
 
diff --git a/src/main/java/com/github/javpower/javavision/util/DjlHandlerUtil.java b/src/main/java/com/github/javpower/javavision/util/DjlHandlerUtil.java
new file mode 100644
index 0000000..88128cb
--- /dev/null
+++ b/src/main/java/com/github/javpower/javavision/util/DjlHandlerUtil.java
@@ -0,0 +1,23 @@
+package com.github.javpower.javavision.util;
+
+import org.apache.commons.lang3.StringUtils;
+
+public class DjlHandlerUtil {
+
+    /**
+     * 获取修复后的模型路径
+     * @param modelPath 如 C:\Users\wosui\AppData\Local\Temp\ocrJava\onnx\image_feature.zip
+     * @return  file:///C:/Users/wosui/AppData/Local/Temp/ocrJava/onnx/image_feature.zip
+     */
+    public static String getFixModelPath(String modelPath){
+        if(StringUtils.isBlank(modelPath)){
+            return "";
+        }
+        StringBuffer path=new StringBuffer();
+        if(!modelPath.startsWith("http")){
+            modelPath=modelPath.replace("\\","/");
+            path.append("file:///").append(modelPath);
+        }
+        return path.toString();
+    }
+}
diff --git a/src/main/java/com/github/javpower/javavision/util/PathConstants.java b/src/main/java/com/github/javpower/javavision/util/PathConstants.java
index e4b09df..b8d2d10 100644
--- a/src/main/java/com/github/javpower/javavision/util/PathConstants.java
+++ b/src/main/java/com/github/javpower/javavision/util/PathConstants.java
@@ -16,7 +16,7 @@ public class PathConstants {
      */
     public static final String NCNN = "/ncnn";
     public static final String ONNX = "/onnx";
-
+    public static final String ONNX_NAME = "onnx";
     /**
      * 模型相关
      **/
-- 
Gitee