Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 12b0b44

Browse files
committed
add data
1 parent b222446 commit 12b0b44

13 files changed

+83
-9
lines changed

.gitignore

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@
22
*_example
33
*_ACDC
44
*__pycache__
5-
# *_slices
6-
# *_volumes
5+
*_slices
6+
*_volumes

code/DMSPS_trainTest_2d.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ CUDA_LAUNCH_BLOCKING=1 CUDA_VISIBLE_DEVICES=0 python train/A_train_weaklySup_SPS
3030
--model unet_cct --exp A_weakly_SPS_2d --fold stage2 --sup_type pseudoLab --trainData trainReT01.txt
3131

3232
# test for stage2
33-
CUDA_LAUNCH_BLOCKING=1 CUDA_VISIBLE_DEVICES=0 python python test/test_2d_forall_fast_txtver.py \
33+
CUDA_LAUNCH_BLOCKING=1 CUDA_VISIBLE_DEVICES=0 python test/test_2d_forall_fast_txtver.py \
3434
--data_root_path /mnt/data/HM/Datasets/ACDC2017/ACDC_for2D \
3535
--model unet_cct --exp A_weakly_SPS_2d --fold stage2
3636

code/test/test_2d_forall_fast_txtver.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,9 @@
3939
parser.add_argument('--model', type=str,
4040
default='unet_cct', help='model_name:unet,unet_cct')
4141
parser.add_argument('--exp', type=str,
42-
default='A3_weakly_PLS_soft_2d', help='Experiment name')
42+
default='A_weakly_SPS_soft_2d', help='Experiment name')
43+
parser.add_argument('--fold', type=str,
44+
default='stage1', help='fold name')
4345
parser.add_argument('--num_classes', type=int, default=4,
4446
help='output channel of network')
4547
parser.add_argument('--patch_size', type=list, default=[256,256],

code/test/test_2d_forall_fast_txtver_forTrainSetUncertaintyOnly_Mean.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@
4040
default='unet_cct', help='model_name:unet,unet_cct, etc. the name of pretrained model from stage1')
4141
parser.add_argument('--exp', type=str,
4242
default='A3_weakly_PLS_soft_2d', help='experiment_name')
43+
parser.add_argument('--fold', type=str,
44+
default='stage1', help='fold name')
4345
parser.add_argument('--num_classes', type=int, default=4,
4446
help='output channel of network')
4547
parser.add_argument('--tt_num', type=int, default=3,

code/test/test_3d_forall_fast_txtver.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,8 @@
4747
default='unet_cct_dropout_3D', help='select model: unet_3D, unet_cct_dropout_3D')
4848
parser.add_argument('--exp', type=str,
4949
default='W_weakly_SPS_3d', help='experiment_name')
50+
parser.add_argument('--fold', type=str,
51+
default='stage1', help='fold name')
5052
parser.add_argument('--num_classes', type=int, default=8,
5153
help='output channel of network')
5254
parser.add_argument('--tt_num', type=int, default=1,

code/test/test_3d_forall_fast_txtver4BraTS.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
parser.add_argument('--exp', type=str,
4949
default='T3_weakly_SPS_3d', help='experiment_name')
5050
parser.add_argument('--fold', type=str,
51-
default='v280_ce_l001_240507', help='fold name')
51+
default='stage1', help='fold name')
5252
parser.add_argument('--num_classes', type=int, default=3,
5353
help='output channel of network')
5454
parser.add_argument('--tt_num', type=int, default=1,

0 commit comments

Comments
 (0)