From 8788c6f7e417a4ac09d98802c2fc8e2f384a30f4 Mon Sep 17 00:00:00 2001 From: cuizhibin Date: Thu, 3 Jul 2025 16:34:43 +0800 Subject: [PATCH] =?UTF-8?q?1.=E6=96=BD=E5=B7=A5=E4=BF=A1=E6=81=AF=E9=97=AE?= =?UTF-8?q?=E9=A2=98=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/dite/znpt/config/YoloModelConfig.java | 45 ------------------- 1 file changed, 45 deletions(-) delete mode 100644 core/src/main/java/com/dite/znpt/config/YoloModelConfig.java 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]); - } -}