Merge branch 'feat/infer_dev_260107' of https://kamco.git.gs.dabeeo.com/MVPTeam/kamco-cd-api into feat/infer_dev_260107
This commit is contained in:
@@ -241,8 +241,7 @@ public class ModelMngService {
|
||||
int endPos = 20;
|
||||
|
||||
List<Basic> 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());
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user