From 6d6ebf8c334e201df668dcb5f63889583b004229 Mon Sep 17 00:00:00 2001 From: "gayoun.park" Date: Fri, 24 Jul 2026 10:02:58 +0900 Subject: [PATCH] =?UTF-8?q?classification=20pt=ED=8C=8C=EC=9D=BC=20?= =?UTF-8?q?=EA=B2=BD=EB=A1=9C=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../inference/InferenceManualApiController.java | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/InferenceManualApiController.java b/src/main/java/com/kamco/cd/kamcoback/inference/InferenceManualApiController.java index 12ad9aef..0c2966eb 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/InferenceManualApiController.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/InferenceManualApiController.java @@ -13,8 +13,11 @@ import io.swagger.v3.oas.annotations.media.Schema; import io.swagger.v3.oas.annotations.responses.ApiResponse; import io.swagger.v3.oas.annotations.responses.ApiResponses; import io.swagger.v3.oas.annotations.tags.Tag; +import java.nio.file.Path; +import java.nio.file.Paths; import java.util.List; import lombok.RequiredArgsConstructor; +import org.springframework.beans.factory.annotation.Value; import org.springframework.data.domain.Page; import org.springframework.web.bind.annotation.*; @@ -27,6 +30,12 @@ public class InferenceManualApiController { private final InferenceManualService inferenceManualService; private final InferenceCommonService inferenceCommonService; + @Value("${file.pt-path}") + private String ptPath; + + @Value("${file.pt-FileName}") + private String ptFileName; + @Operation(summary = "추론 결과로 추론 목록 및 shp 생성", description = "추론 결과로 추론 목록 및 shp 생성") @ApiResponses( value = { @@ -92,10 +101,14 @@ public class InferenceManualApiController { @Parameter(description = "batch_ids", example = "2026,2027") @RequestParam List batchIds) { + String defaultPath = ptPath; + String defaultFileName = ptFileName; + Path ptPath = Paths.get(defaultPath, defaultFileName); + InferenceClassificationDto.ClassificationRequestDto request = new InferenceClassificationDto.ClassificationRequestDto(); request.setBatch_ids(batchIds); - request.setCls_model_path("/data/ckpt/model/v6-cls-checkpoints/yolov8_6th-6m.pt"); + request.setCls_model_path(ptPath.toString()); request.setCls_model_version("testing-re-cls"); // Classification 실행 api 호출