PerceptionComputingLab
diff --git a/‎CETUS/test/patient03.pkl‎
32 MB b/‎CETUS/test/patient03.pkl‎
32 MB
diff --git a/‎CETUS/train/patient01.pkl‎
32 MB b/‎CETUS/train/patient01.pkl‎
32 MB
diff --git a/‎CETUS/valid/patient16.pkl‎
32 MB b/‎CETUS/valid/patient16.pkl‎
32 MB
diff --git a/‎Model/__pycache__/config.cpython-39.pyc‎
689 Bytes b/‎Model/__pycache__/config.cpython-39.pyc‎
689 Bytes
diff --git a/‎Model/__pycache__/dataset.cpython-39.pyc‎
1.02 KB b/‎Model/__pycache__/dataset.cpython-39.pyc‎
1.02 KB
diff --git a/‎Model/__pycache__/losses.cpython-37.pyc‎
2.96 KB b/‎Model/__pycache__/losses.cpython-37.pyc‎
2.96 KB
diff --git a/‎Model/__pycache__/losses.cpython-39.pyc‎
4.13 KB b/‎Model/__pycache__/losses.cpython-39.pyc‎
4.13 KB
diff --git a/‎Model/__pycache__/model.cpython-37.pyc‎
5.06 KB b/‎Model/__pycache__/model.cpython-37.pyc‎
5.06 KB
diff --git a/‎Model/__pycache__/model.cpython-39.pyc‎
7.79 KB b/‎Model/__pycache__/model.cpython-39.pyc‎
7.79 KB
diff --git a/‎Model/config.py‎
Lines changed: 18 additions & 0 deletions b/‎Model/config.py‎
Lines changed: 18 additions & 0 deletions
@@ -0,0 +1,18 @@
+class Config:
+    gpu = '0'
+    data_path = './CETUS/'
+    model = 'vm2'
+    result_dir = './Result'
+    lr = 1e-5
+    max_epochs = 201
+    sim_loss = 'mse'
+    alpha = 0.04
+    batch_size = 1
+    n_save_epoch = 5
+    model_dir = './Checkpoint/exp_'
+    log_dir = './Log'
+    saved_unet_name = 'unet_model.pth'
+    saved_tnet_name = 'tnet_model.pth'
+
+    # test时参数
+    checkpoint_path = "./Checkpoint/exp_1715837905.2272804/"