feat: class name
This commit is contained in:
@@ -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;
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user