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

모델등록 수정

Browse Source
This commit is contained in:
teddy
2026-01-27 17:48:46 +09:00
parent 5e65474a39
commit db5325a1fa
1 changed files with 34 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -15,6 +15,7 @@ import com.kamco.cd.kamcoback.upload.service.UploadService;
import java.io.IOException; import java.io.IOException;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.LocalDate; import java.time.LocalDate;
import java.util.List; import java.util.List;
import java.util.UUID; import java.util.UUID;
@@ -237,22 +238,43 @@ public class ModelMngService {
FIleChecker.getFilesFromAllDepth( FIleChecker.getFilesFromAllDepth(
dirPath, "*", "pth,py,pt,json", 10, "name", startPos, endPos); dirPath, "*", "pth,py,pt,json", 10, "name", startPos, endPos);
boolean hasPt = false; // pt 파일 존재 여부
for (Basic dto : files) { for (Basic dto : files) {
// 예: 파일명 출력 및 추가 작업 // 예: 파일명 출력 및 추가 작업
String foldNm = dto.getFullPath().replace(dto.getFileNm(), ""); String foldNm = dto.getFullPath().replace(dto.getFileNm(), "");
if (dto.getExtension().equals("pth")) { switch (dto.getExtension()) {
modelUploadResDto.setCdModelPath(foldNm); case "pth" -> {
modelUploadResDto.setCdModelFileName(dto.getFileNm()); modelUploadResDto.setCdModelPath(foldNm);
} else if (dto.getExtension().equals("py")) { modelUploadResDto.setCdModelFileName(dto.getFileNm());
modelUploadResDto.setCdModelConfigPath(foldNm); }
modelUploadResDto.setCdModelConfigFileName(dto.getFileNm()); case "py" -> {
} else if (dto.getExtension().equals("pt")) { modelUploadResDto.setCdModelConfigPath(foldNm);
modelUploadResDto.setClsModelPath(foldNm); modelUploadResDto.setCdModelConfigFileName(dto.getFileNm());
modelUploadResDto.setClsModelFileName(dto.getFileNm()); }
} else if (dto.getExtension().equals("json")) { case "pt" -> {
modelUploadResDto.setJsonPath(foldNm); modelUploadResDto.setClsModelPath(foldNm);
modelUploadResDto.setJsonFileName(dto.getFileNm()); modelUploadResDto.setClsModelFileName(dto.getFileNm());
hasPt = true;
}
case "json" -> {
modelUploadResDto.setJsonPath(foldNm);
modelUploadResDto.setJsonFileName(dto.getFileNm());
}
}
}
if (!hasPt) {
String defaultPath = "/kamco-nfs/ckpt/classification/";
String defaultFileName = "v5-best.pt";
Path ptPath = Paths.get(defaultPath, defaultFileName);
if (Files.exists(ptPath)) {
modelUploadResDto.setClsModelPath(defaultPath);
modelUploadResDto.setClsModelFileName(defaultFileName);
} }
} }