[KC-103] 추론 실행 배치 오류 수정
This commit is contained in:
@@ -54,7 +54,7 @@ public class MapSheetInferenceJobService {
|
||||
public void runBatch() {
|
||||
|
||||
if ("local".equalsIgnoreCase(profile)) {
|
||||
return;
|
||||
// return;
|
||||
}
|
||||
|
||||
try {
|
||||
@@ -149,6 +149,16 @@ public class MapSheetInferenceJobService {
|
||||
InferenceProgressDto progressDto =
|
||||
inferenceResultCoreService.getInferenceAiResultById(id, type, modelUuid);
|
||||
|
||||
String inferenceType = "";
|
||||
|
||||
if (type.equals("M1")) {
|
||||
inferenceType = "G1";
|
||||
} else if (type.equals("M2")) {
|
||||
inferenceType = "G2";
|
||||
} else if (type.equals("M3")) {
|
||||
inferenceType = "G3";
|
||||
}
|
||||
|
||||
pred_requests_areas predRequestsAreas = new pred_requests_areas();
|
||||
predRequestsAreas.setInput1_year(progressDto.getPred_requests_areas().getInput1_year());
|
||||
predRequestsAreas.setInput2_year(progressDto.getPred_requests_areas().getInput2_year());
|
||||
@@ -158,6 +168,7 @@ public class MapSheetInferenceJobService {
|
||||
progressDto.getPred_requests_areas().getInput2_scene_path());
|
||||
|
||||
InferenceSendDto m = new InferenceSendDto();
|
||||
m.setPred_requests_areas(predRequestsAreas);
|
||||
m.setModel_version(progressDto.getModelVersion());
|
||||
m.setCd_model_path(progressDto.getCdModelPath() + "/" + progressDto.getCdModelFileName());
|
||||
m.setCd_model_config(
|
||||
@@ -165,7 +176,7 @@ public class MapSheetInferenceJobService {
|
||||
m.setCls_model_path(
|
||||
progressDto.getCdModelClsPath() + "/" + progressDto.getCdModelClsFileName());
|
||||
m.setCls_model_version(progressDto.getClsModelVersion());
|
||||
m.setCd_model_type(type);
|
||||
m.setCd_model_type(inferenceType);
|
||||
m.setPriority(progressDto.getPriority());
|
||||
|
||||
// 추론 다음모델 실행
|
||||
@@ -195,6 +206,10 @@ public class MapSheetInferenceJobService {
|
||||
|
||||
// TODO 추후 삭제
|
||||
if ("local".equals(profile)) {
|
||||
if (dto.getPred_requests_areas() == null) {
|
||||
dto.setPred_requests_areas(new InferenceSendDto.pred_requests_areas());
|
||||
}
|
||||
|
||||
dto.getPred_requests_areas().setInput1_scene_path("/kamco-nfs/requests/2023_local.geojson");
|
||||
dto.getPred_requests_areas().setInput2_scene_path("/kamco-nfs/requests/2024_local.geojson");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user