test

kamco-make-dataset-generation/src/main/resources/application-prod.yml

@@ -1,6 +1,6 @@
 spring:
   datasource:
-    url: jdbc:postgresql://127.0.0.1:15432/kamco_cds
+    url: jdbc:postgresql://172.16.4.56:15432/kamco_cds
     username: kamco_cds
     password: kamco_cds_Q!W@E#R$
     hikari:
@@ -10,22 +10,18 @@ spring:
     job:
       name: exportGeoJsonJob  # 기본 실행 Job 지정
 
-training-data:
-  geojson-dir: /kamco-nfs/dataset
-
 # Train Model Docker Configuration
-train-data:
+training-data:
+  geojson-dir: /data/dataset
   docker:
-    image: kamco-cd-train:latest
-    data-volume: /kamco-nfs/dataset:/data
-    checkpoints-volume: /kamco-nfs/checkpoints:/checkpoints
-    dataset-folder: /data/dataset
-    output-folder: /data/output
-    input-size: "512"
-    crop-size: "256"
-    batch-size: 8
-    gpu-ids: "0,1,2,3"
-    gpus: 4
-    lr: "0.001"
-    backbone: resnet50
-    epochs: 100
+    dataset-volume: /data/dataset:/dataset
+    images-volume: /data/images:/kamco-nfs:ro
+    input-root: /dataset
+    output-root: /dataset
+    patch-size: 512
+    overlap-pct: 50
+    train-val-test-ratio:
+      - "0.7"
+      - "0.2"
+      - "0.1"
+    keep-empty-ratio: 0.1

kamco-make-dataset-generation/src/main/resources/application.yml

@@ -29,8 +29,8 @@ training-data:
   docker:
     image: kamco-cd-dataset:latest
     user: "1000:1000"
-    dataset-volume: /kamco-nfs/dataset:/dataset
-    images-volume: /kamco-nfs/images:/kamco-nfs:ro
+    dataset-volume: /data/dataset:/dataset
+    images-volume: /data/images:/kamco-nfs:ro
     input-root: /dataset
     output-root: /dataset
     patch-size: 512