diff --git a/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/config/DockerProperties.java b/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/config/DockerProperties.java index fc834c1..0701287 100644 --- a/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/config/DockerProperties.java +++ b/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/config/DockerProperties.java @@ -13,6 +13,7 @@ public class DockerProperties { private String image; private String user; private String datasetVolume; + private String outputVolume; private String imagesVolume; private String inputRoot; private String outputRoot; @@ -20,4 +21,5 @@ public class DockerProperties { private int overlapPct; private List trainValTestRatio; private double keepEmptyRatio; + private double keepEmtpyRatioVal; } diff --git a/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/service/DockerRunnerService.java b/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/service/DockerRunnerService.java index a679659..e1d07c4 100644 --- a/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/service/DockerRunnerService.java +++ b/kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/service/DockerRunnerService.java @@ -161,11 +161,16 @@ public class DockerRunnerService { // 볼륨 마운트 (호스트:컨테이너) cmd.add("-v"); - cmd.add(dockerProperties.getDatasetVolume()); // 예: "/kamco-nfs/dataset:/dataset" + cmd.add(dockerProperties.getDatasetVolume()); log.debug(" - Dataset Volume: {}", dockerProperties.getDatasetVolume()); + //output 마운트 cmd.add("-v"); - cmd.add(dockerProperties.getImagesVolume()); // 예: "/kamco-nfs/images:/images" + cmd.add(dockerProperties.getOutputVolume()); + log.debug(" - Output Volume: {}", dockerProperties.getOutputVolume()); + + cmd.add("-v"); + cmd.add(dockerProperties.getImagesVolume()); log.debug(" - Images Volume: {}", dockerProperties.getImagesVolume()); // Entrypoint 및 이미지 @@ -209,6 +214,10 @@ public class DockerRunnerService { cmd.add(String.valueOf(dockerProperties.getKeepEmptyRatio())); log.debug(" - Keep Empty Ratio: {}", dockerProperties.getKeepEmptyRatio()); + cmd.add("--keep_empty_ratio_val"); + cmd.add(String.valueOf(dockerProperties.getKeepEmptyRatioVal())); + log.debug(" - Keep Empty Ratio Val: {}", dockerProperties.getKeepEmptyRatioVal()); + log.debug("Docker 명령어 파라미터 구성 완료"); return cmd; } diff --git a/kamco-make-dataset-generation/src/main/resources/application.yml b/kamco-make-dataset-generation/src/main/resources/application.yml index 787eaf3..879120f 100644 --- a/kamco-make-dataset-generation/src/main/resources/application.yml +++ b/kamco-make-dataset-generation/src/main/resources/application.yml @@ -28,14 +28,16 @@ training-data: docker: image: kamco-cd-dataset:latest user: "1000:1000" - dataset-volume: /data/dataset:/dataset - images-volume: /data/images:/kamco-nfs:ro - input-root: /dataset - output-root: /dataset + dataset-volume: /data/dataset:/data/labels:ro + output-volume: /data/dataset:/data/out + images-volume: /data/images:/data/images + input-root: /data/labels + output-root: /data/out patch-size: 512 overlap-pct: 50 train-val-test-ratio: - "0.7" - "0.2" - "0.1" - keep-empty-ratio: 0.1 \ No newline at end of file + keep-empty-ratio: 0.1 + keep-empty-ratio-val: 0.3 \ No newline at end of file