From 87374c8f6fe12136940e89d62ddabff094359caa Mon Sep 17 00:00:00 2001 From: teddy Date: Thu, 29 Jan 2026 12:08:01 +0900 Subject: [PATCH] =?UTF-8?q?=EB=AA=A8=EB=8D=B8=EB=93=B1=EB=A1=9D=20pt=20?= =?UTF-8?q?=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../kamco/cd/kamcoback/model/service/ModelMngService.java | 8 +------- src/main/resources/application-dev.yml | 4 ++-- src/main/resources/application-prod.yml | 4 ++-- 3 files changed, 5 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java index 45bbd8db..eb4e9df9 100644 --- a/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java +++ b/src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java @@ -241,8 +241,7 @@ public class ModelMngService { int endPos = 20; List files = - FIleChecker.getFilesFromAllDepth( - dirPath, "*", "pth,pt,py,json", 10, "name", startPos, endPos); + FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,json", 10, "name", startPos, endPos); boolean hasPt = false; // pt 파일 존재 여부 @@ -259,11 +258,6 @@ public class ModelMngService { modelUploadResDto.setCdModelConfigPath(foldNm); modelUploadResDto.setCdModelConfigFileName(dto.getFileNm()); } - case "pt" -> { - modelUploadResDto.setClsModelPath(foldNm); - modelUploadResDto.setClsModelFileName(dto.getFileNm()); - hasPt = true; - } case "json" -> { modelUploadResDto.setJsonPath(foldNm); modelUploadResDto.setJsonFileName(dto.getFileNm()); diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml index 60a7057f..e99d8520 100644 --- a/src/main/resources/application-dev.yml +++ b/src/main/resources/application-dev.yml @@ -104,8 +104,8 @@ file: model-tmp-dir: ${file.model-dir}tmp/ model-file-extention: pth,json,py - pt-path: /kamco-nfs/ckpt/classification/ - pt-FileName: v5-best.pt + pt-path: /kamco-nfs/ckpt/model/v6-cls-checkpoints/ + pt-FileName: yolov8_6th-6m.pt inference: url: http://10.100.0.11:8000/jobs diff --git a/src/main/resources/application-prod.yml b/src/main/resources/application-prod.yml index 02c833f1..9b8819bf 100644 --- a/src/main/resources/application-prod.yml +++ b/src/main/resources/application-prod.yml @@ -59,8 +59,8 @@ file: model-tmp-dir: ${file.model-dir}tmp/ model-file-extention: pth,json,py - pt-path: /kamco-nfs/ckpt/classification/ - pt-FileName: v5-best.pt + pt-path: /kamco-nfs/ckpt/model/v6-cls-checkpoints/ + pt-FileName: yolov8_6th-6m.pt inference: url: http://10.100.0.11:8000/jobs