From 79e645e51f1895032ce4891d4927c66270d746f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?dean=5B=EB=B0=B1=EB=B3=91=EB=82=A8=5D?= Date: Tue, 2 Dec 2025 14:37:34 +0900 Subject: [PATCH] feat: class name --- .../common/enums/DetectionClassification.java | 20 +++++++++---------- .../inference/dto/InferenceResultDto.java | 8 +++++++- 2 files changed, 17 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/kamco/cd/kamcoback/common/enums/DetectionClassification.java b/src/main/java/com/kamco/cd/kamcoback/common/enums/DetectionClassification.java index a3a2bd7c..ea806e1a 100644 --- a/src/main/java/com/kamco/cd/kamcoback/common/enums/DetectionClassification.java +++ b/src/main/java/com/kamco/cd/kamcoback/common/enums/DetectionClassification.java @@ -6,20 +6,20 @@ import lombok.Getter; @Getter @AllArgsConstructor public enum DetectionClassification { - BUILDING("building", "빌딩"), - CONTAINER("container", "컨테이너(창고·적재함)"), - FIELD("field", "경작지 / 들판"), + BUILDING("building", "건물"), + CONTAINER("container", "컨테이너"), + FIELD("field", "경작지"), FOREST("forest", "숲"), - GRASS("grass", "초지 / 잔디지역"), + GRASS("grass", "초지"), GREENHOUSE("greenhouse", "비닐하우스"), - LAND("land", "나지"), + LAND("land", "일반토지"), ORCHARD("orchard", "과수원"), ROAD("road", "도로"), - STONE("stone", "암석 / 돌 지역"), - TANK("tank", "탱크(저유탱크 등 저장시설)"), - TUMULUS("tumulus", "고분(무덤)"), - WASTE("waste", "폐기물 적치장 / 황폐지"), - WATER("water", "수체(水域) / 물"), + STONE("stone", "모래/자갈"), + TANK("tank", "물탱크"), + TUMULUS("tumulus", "토분(무덤)"), + WASTE("waste", "폐기물"), + WATER("water", "물"), ETC("ETC", "기타"); // For 'etc' (miscellaneous/other) private final String id; diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java index 9f5cadf1..224dbaad 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceResultDto.java @@ -196,11 +196,17 @@ public class InferenceResultDto { // classification info @Getter - @AllArgsConstructor public static class Clazz { private String code; + private String name; private Double score; + + public Clazz(String code, Double score) { + this.code = code; + this.score = score; + this.name = DetectionClassification.fromString(code).getDesc(); + } } // 좌표 정보 point