diff --git a/core/src/main/java/com/dite/znpt/config/YoloModelConfig.java b/core/src/main/java/com/dite/znpt/config/YoloModelConfig.java deleted file mode 100644 index 7e92b49..0000000 --- a/core/src/main/java/com/dite/znpt/config/YoloModelConfig.java +++ /dev/null @@ -1,45 +0,0 @@ -package com.dite.znpt.config; - -import ai.onnxruntime.OrtEnvironment; -import ai.onnxruntime.OrtException; -import ai.onnxruntime.OrtSession; -import lombok.Getter; -import org.springframework.context.annotation.Configuration; - -import javax.annotation.PostConstruct; -import java.util.ArrayList; -import java.util.List; -import java.util.Random; -import java.util.regex.Matcher; -import java.util.regex.Pattern; - -@Getter -@Configuration -public class YoloModelConfig { - - private OrtEnvironment env; - private OrtSession session; - private String[] labels; - private List colors; - - @PostConstruct - public void init() throws OrtException { - env = OrtEnvironment.getEnvironment(); - OrtSession.SessionOptions options = new OrtSession.SessionOptions(); - session = env.createSession("d:\\tmp\\best.onnx", options); - - // 解析 labels - String names = session.getMetadata().getCustomMetadata().get("names"); - Pattern pattern = Pattern.compile("'([^']*)'"); - Matcher matcher = pattern.matcher(names); - List labelList = new ArrayList<>(); - colors = new ArrayList<>(); - - Random random = new Random(); - while (matcher.find()) { - labelList.add(matcher.group(1)); - colors.add(new double[]{random.nextDouble()*256, random.nextDouble()*256, random.nextDouble()*256}); - } - labels = labelList.toArray(new String[0]); - } -}