feat: class name

This commit is contained in:
2025-12-02 14:37:34 +09:00
parent 60e962069d
commit 79e645e51f
2 changed files with 17 additions and 11 deletions

View File

@@ -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;

View File

@@ -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