Merge branch 'feat/infer_dev_260107' of https://10.100.0.10:3210/dabeeo/kamco-dabeeo-backoffice into feat/infer_dev_260107

This commit is contained in:
2026-01-27 17:33:43 +09:00
2 changed files with 15 additions and 6 deletions

View File

@@ -228,6 +228,8 @@ public class ModelMngDto {
private String clsModelFileName;
private int chunkIndex;
private int chunkTotalIndex;
@JsonIgnore private String jsonPath;
@JsonIgnore private String jsonFileName;
public double getUploadRate() {
if (chunkIndex < 0 || chunkTotalIndex < 0) {

View File

@@ -95,18 +95,21 @@ public class ModelMngService {
|| modelUploadResDto.getCdModelConfigFileName() == null
|| modelUploadResDto.getCdModelConfigFileName().isEmpty()
|| modelUploadResDto.getClsModelFileName() == null
|| modelUploadResDto.getClsModelFileName().isEmpty()) {
|| modelUploadResDto.getClsModelFileName().isEmpty()
|| modelUploadResDto.getJsonFileName() == null
|| modelUploadResDto.getJsonFileName().isEmpty()) {
log.warn(
"model file missing: uuid={}, path={}, cdModel={}, cdConfig={}, clsModel={}",
"model file missing: uuid={}, path={}, cdModel={}, cdConfig={}, clsModel={}, jsonFileName={}",
uuid,
addReq.getFilePath(),
modelUploadResDto.getCdModelFileName(),
modelUploadResDto.getCdModelConfigFileName(),
modelUploadResDto.getClsModelFileName());
modelUploadResDto.getClsModelFileName(),
modelUploadResDto.getJsonFileName());
return new ApiResponseDto.ResponseObj(
ApiResponseCode.NOT_FOUND, "pth,pt,py파일이 모두 존재하지 않습니다");
ApiResponseCode.NOT_FOUND, "pth,pt,py,json파일이 모두 존재하지 않습니다");
}
addReq.setCdModelPath(modelUploadResDto.getCdModelPath());
@@ -142,7 +145,7 @@ public class ModelMngService {
modelMetricAddReq.setIou(0);
ObjectMapper mapper = new ObjectMapper();
String filePath = modelUploadResDto.getClsModelPath() + modelUploadResDto.getClsModelFileName();
String filePath = modelUploadResDto.getJsonPath() + modelUploadResDto.getJsonFileName();
String dataJson = null;
try {
dataJson = Files.readString(Path.of(filePath));
@@ -231,7 +234,8 @@ public class ModelMngService {
int endPos = 20;
List<Basic> files =
FIleChecker.getFilesFromAllDepth(dirPath, "*", "pth,py,pt", 10, "name", startPos, endPos);
FIleChecker.getFilesFromAllDepth(
dirPath, "*", "pth,py,pt,json", 10, "name", startPos, endPos);
for (Basic dto : files) {
// 예: 파일명 출력 및 추가 작업
@@ -246,6 +250,9 @@ public class ModelMngService {
} else if (dto.getExtension().equals("pt")) {
modelUploadResDto.setClsModelPath(foldNm);
modelUploadResDto.setClsModelFileName(dto.getFileNm());
} else if (dto.getExtension().equals("json")) {
modelUploadResDto.setJsonPath(foldNm);
modelUploadResDto.setJsonFileName(dto.getFileNm());
}
}