MVPTeam/kamco-cd-cron
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

docker 명령어 최신 버전으로 수정

Browse Source
This commit is contained in:
Gina(박가연)
2026-05-08 15:07:07 +09:00
parent c1430da2f9
commit e6b3b1d8cf
3 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/config/DockerProperties.java
View File

@@ -13,6 +13,7 @@ public class DockerProperties {
private String image; private String image;
private String user; private String user;
private String datasetVolume; private String datasetVolume;
private String outputVolume;
private String imagesVolume; private String imagesVolume;
private String inputRoot; private String inputRoot;
private String outputRoot; private String outputRoot;
@@ -20,4 +21,5 @@ public class DockerProperties {
private int overlapPct; private int overlapPct;
private List<String> trainValTestRatio; private List<String> trainValTestRatio;
private double keepEmptyRatio; private double keepEmptyRatio;
private double keepEmtpyRatioVal;
} }

13
kamco-make-dataset-generation/src/main/java/com/kamco/cd/geojsonscheduler/service/DockerRunnerService.java
View File

@@ -161,11 +161,16 @@ public class DockerRunnerService {
// 볼륨 마운트 (호스트:컨테이너) // 볼륨 마운트 (호스트:컨테이너)
cmd.add("-v"); cmd.add("-v");
cmd.add(dockerProperties.getDatasetVolume()); // 예: "/kamco-nfs/dataset:/dataset" cmd.add(dockerProperties.getDatasetVolume());
log.debug(" - Dataset Volume: {}", dockerProperties.getDatasetVolume()); log.debug(" - Dataset Volume: {}", dockerProperties.getDatasetVolume());
//output 마운트
cmd.add("-v"); 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()); log.debug(" - Images Volume: {}", dockerProperties.getImagesVolume());
// Entrypoint 및 이미지 // Entrypoint 및 이미지
@@ -209,6 +214,10 @@ public class DockerRunnerService {
cmd.add(String.valueOf(dockerProperties.getKeepEmptyRatio())); cmd.add(String.valueOf(dockerProperties.getKeepEmptyRatio()));
log.debug(" - Keep Empty Ratio: {}", 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 명령어 파라미터 구성 완료"); log.debug("Docker 명령어 파라미터 구성 완료");
return cmd; return cmd;
} }

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

@@ -28,10 +28,11 @@ training-data:
docker: docker:
image: kamco-cd-dataset:latest image: kamco-cd-dataset:latest
user: "1000:1000" user: "1000:1000"
dataset-volume: /data/dataset:/dataset dataset-volume: /data/dataset:/data/labels:ro
images-volume: /data/images:/kamco-nfs:ro output-volume: /data/dataset:/data/out
input-root: /dataset images-volume: /data/images:/data/images
output-root: /dataset input-root: /data/labels
output-root: /data/out
patch-size: 512 patch-size: 512
overlap-pct: 50 overlap-pct: 50
train-val-test-ratio: train-val-test-ratio:
@@ -39,3 +40,4 @@ training-data:
- "0.2" - "0.2"
- "0.1" - "0.1"
keep-empty-ratio: 0.1 keep-empty-ratio: 0.1
keep-empty-ratio-val: 0.3