init

liujiyaoFDU · Aug 18, 2023 · c071937 · c071937
1 parent cd4aca4
commit c071937
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 43 deletions.
diff --git a/Model/CBAM.py b/Model/CBAM.py
@@ -1,4 +1,4 @@
-"""CBAM:通道+空间注意力机制
+"""CBAM
 
 author:jiyao liu
 """

diff --git a/Model/Pix2Pix_v6.py b/Model/Pix2Pix_v6.py
@@ -1,7 +1,6 @@
 """
-author:jiyao liu 20220223
+author:jiyao liu 20230223
 
-Down: 1. 增加mask分支;
 """
 
 import torch.nn.functional as F

diff --git a/Yaml/mrgan.yaml b/Yaml/mrgan.yaml
@@ -1,38 +1,24 @@
-#       --p2pv2--
-#   data: Liver_Fibrosis_PMX_REG_transformed5
-#         loaded by datasets_with_masked_image.py
-#   input:  {'A': stack_input.float(), 
-#         'B': transformed_images[4].float(), 
-#         'C': masked_stack_input.float(),
-#         'D': masked_transformed_images[4].float()}
-#   aug: source and target same randomaffine
-#   txt: jiyaoliu/Projects/Liver_PMX/Reg-GAN-PMX/data
-#   model: Pix2Pix_v2.py (add share bottleneck model )
-#   Trainer: p2pTrainer_v2.py 
-
-
-## 必须修改内容
-name: mrgan1 # 保存visdom环境名
+name: mrgan # visdom env name
 root: ./
-save_root: './output/v1.0/'  # save path of training
-image_save: './output/v1.0/img_val/'  # save path of test result
+save_root: './output/v1.1/'  # save path of training
+image_save: './output/v1.1/img_val/'  # save path of test result
 # lamda weight
 Adv_lamda: 1
 P2P_lamda: 5
 blur_lamda: 5
-perceptual: 0.5 # 感知损失权重
+perceptual: 0.5 
 Corr_lamda: 1
 Smooth_lamda: 1
 shape : 1
 
 
-## 可以修改的内容
-port: 6019   #  port parameters
-num_classes : 2  # 分割的类别(前景+背景)
-palette : [[0], [1]]  # mask颜色表
+## 
+port: 6019   # visdom port parameters
+num_classes : 2  # class num of seg (foreground and background)
+palette : [[0], [1]]  # mask color table
 unet_chk : './chk/UNet.pth'
 
-## 不变内容
+## 
 cuda: True
 resize: 256
 regist : True

diff --git a/trainer/datasets_unet.py b/trainer/datasets_unet.py
@@ -1,13 +1,6 @@
 #!/usr/bin/python3
 """
-author:jiyao liu 20220223
-DONE:1.输入增加 ROI mask:
-        return {'A': stack_input.float(), 
-        'B': transformed_images[4].float(), 
-        'C': masked_stack_input.float(),
-        'D': masked_transformed_images[4].float()}
-
-DONE:2. 数据增广: source和target随机相同仿射变换
+author:jiyao liu 20230223
 
 DONE:
 

diff --git a/trainer/datasets_v6.py b/trainer/datasets_v6.py
@@ -1,15 +1,7 @@
 #!/usr/bin/python3
 """
-author:jiyao liu 20220223
-DONE:1.输入增加 ROI mask:
-        return {'A': stack_input.float(), 
-        'B': transformed_images[4].float(), 
-        'C': masked_stack_input.float(),
-        'D': masked_transformed_images[4].float()}
+author:jiyao liu 20230223
 
-DONE:2. 数据增广: source和target随机相同仿射变换
-
-DONE:
 
 """