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