docker 명령어 최신 버전으로 수정
This commit is contained in:
@@ -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<String> trainValTestRatio;
|
||||
private double keepEmptyRatio;
|
||||
private double keepEmtpyRatioVal;
|
||||
}
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
@@ -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
|
||||
keep-empty-ratio: 0.1
|
||||
keep-empty-ratio-val: 0.3
|
||||
Reference in New Issue
Block a user