diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceSendDto.java b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceSendDto.java index 5f7ca4fc..e4783f00 100644 --- a/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceSendDto.java +++ b/src/main/java/com/kamco/cd/kamcoback/inference/dto/InferenceSendDto.java @@ -23,6 +23,14 @@ public class InferenceSendDto { private String cd_model_type; private Double priority; + public String getCd_model_path() { + return cd_model_path==null?null:cd_model_path.replace("kamcd-nfs", "data"); + } + + public String getCls_model_path() { + return cls_model_path==null?null:cls_model_path.replace("kamcd-nfs", "data"); + } + @Getter @Setter @AllArgsConstructor @@ -34,5 +42,14 @@ public class InferenceSendDto { private Integer input2_year; private String input1_scene_path; private String input2_scene_path; + + public String getInput1_scene_path() { + + return input1_scene_path==null?null:input1_scene_path.replace("kamcd-nfs", "data"); + } + + public String getInput2_scene_path() { + return input2_scene_path==null?null:input2_scene_path.replace("kamcd-nfs", "data"); + } } }