MVPTeam/kamco-cd-api
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'feat/infer_dev_260107' of https://kamco.git.gs.dabeeo.com/MVPTeam/kamco-cd-api into feat/infer_dev_260107

Browse Source
This commit is contained in:
Gina(박가연)
2026-01-29 12:15:45 +09:00
parent 7f60a63a0f 87374c8f6f
commit d83f8befa9
3 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/com/kamco/cd/kamcoback/model/service/ModelMngService.java
View File

@@ -241,8 +241,7 @@ public class ModelMngService {
int endPos = 20; int endPos = 20;
List<Basic> files = List<Basic> files =
FIleChecker.getFilesFromAllDepth( FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,json", 10, "name", startPos, endPos);
dirPath, "*", "pth,pt,py,json", 10, "name", startPos, endPos);
boolean hasPt = false; // pt 파일 존재 여부 boolean hasPt = false; // pt 파일 존재 여부
@@ -259,11 +258,6 @@ public class ModelMngService {
modelUploadResDto.setCdModelConfigPath(foldNm); modelUploadResDto.setCdModelConfigPath(foldNm);
modelUploadResDto.setCdModelConfigFileName(dto.getFileNm()); modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
} }
case "pt" -> {
modelUploadResDto.setClsModelPath(foldNm);
modelUploadResDto.setClsModelFileName(dto.getFileNm());
hasPt = true;
}
case "json" -> { case "json" -> {
modelUploadResDto.setJsonPath(foldNm); modelUploadResDto.setJsonPath(foldNm);
modelUploadResDto.setJsonFileName(dto.getFileNm()); modelUploadResDto.setJsonFileName(dto.getFileNm());

4
src/main/resources/application-dev.yml
View File

@@ -104,8 +104,8 @@ file:
model-tmp-dir: ${file.model-dir}tmp/ model-tmp-dir: ${file.model-dir}tmp/
model-file-extention: pth,json,py model-file-extention: pth,json,py
pt-path: /kamco-nfs/ckpt/classification/ pt-path: /kamco-nfs/ckpt/model/v6-cls-checkpoints/
pt-FileName: v5-best.pt pt-FileName: yolov8_6th-6m.pt
inference: inference:
url: http://10.100.0.11:8000/jobs url: http://10.100.0.11:8000/jobs

4
src/main/resources/application-prod.yml
View File

@@ -59,8 +59,8 @@ file:
model-tmp-dir: ${file.model-dir}tmp/ model-tmp-dir: ${file.model-dir}tmp/
model-file-extention: pth,json,py model-file-extention: pth,json,py
pt-path: /kamco-nfs/ckpt/classification/ pt-path: /kamco-nfs/ckpt/model/v6-cls-checkpoints/
pt-FileName: v5-best.pt pt-FileName: yolov8_6th-6m.pt
inference: inference:
url: http://10.100.0.11:8000/jobs url: http://10.100.0.11:8000/jobs