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 e1d07c4..cd2e485 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 @@ -184,11 +184,11 @@ public class DockerRunnerService { // 입출력 폴더 설정 cmd.add("--labelling-folder"); - cmd.add("request/" + resultUid); + cmd.add(resultUid); log.debug(" - Labelling Folder: request/{}", resultUid); cmd.add("--output-folder"); - cmd.add("response/" + resultUid); + cmd.add(resultUid); log.debug(" - Output Folder: response/{}", resultUid); // 파이프라인 파라미터 diff --git a/kamco-make-dataset-generation/src/main/resources/application-dev.yml b/kamco-make-dataset-generation/src/main/resources/application-dev.yml index 7f2bfb8..38dd104 100644 --- a/kamco-make-dataset-generation/src/main/resources/application-dev.yml +++ b/kamco-make-dataset-generation/src/main/resources/application-dev.yml @@ -12,3 +12,19 @@ spring: training-data: geojson-dir: /data/dataset + docker: + image: kamco-cd-dataset:latest + user: "1000:1000" + dataset-volume: /data/dataset/request:/data/labels:ro + output-volume: /data/dataset/response:/data/out + images-volume: /data/images:/kamco-nfs:ro + 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 + keep-empty-ratio-val: 0.3 \ No newline at end of file diff --git a/kamco-make-dataset-generation/src/main/resources/application-prod.yml b/kamco-make-dataset-generation/src/main/resources/application-prod.yml index 679fa1d..9cdd4c6 100644 --- a/kamco-make-dataset-generation/src/main/resources/application-prod.yml +++ b/kamco-make-dataset-generation/src/main/resources/application-prod.yml @@ -14,14 +14,18 @@ spring: training-data: geojson-dir: /data/dataset docker: - dataset-volume: /data/dataset:/dataset - images-volume: /data/images:/kamco-nfs:ro - input-root: /dataset - output-root: /dataset + image: kamco-cd-dataset:latest + user: "1000:1000" + dataset-volume: /data/dataset/request:/data/labels:ro + output-volume: /data/dataset/response:/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