update

JuneHM · JuneHM · commit 7b028426eae0 · 2025-04-02T11:59:27.000+08:00
diff --git a/code/test/test_3d_forall_fast_txtver.py b/code/test/test_3d_forall_fast_txtver.py
@@ -23,7 +23,7 @@
 
 # from networks.efficientunet import UNet
 from networks.net_factory_3d import net_factory_3d
-from test.uttils import calculate_metric_percase, logInference, get_the_first_k_largest_components, get_rgb_from_uncertainty
+from uttils import calculate_metric_percase, logInference, get_the_first_k_largest_components, get_rgb_from_uncertainty
 
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
@@ -126,7 +126,7 @@ def pred_single_case_3d(net, net_type, image, stride_xy, stride_z, patch_size, n
                 test_patch = torch.from_numpy(test_patch).cuda()
                 with torch.no_grad():
                     if net_type == "unet_cct_dropout_3D" or net_type == "attention_unet_2dual_3d"\
-                        or net_type == "unetr_2dual_3d":
+                        or net_type == "unetr_2dual_3d" or net_type =="unet_cct_dp_3D":
                         y_main, _ = net(test_patch)
                     else:
                         y_main = net(test_patch)
diff --git a/code/train/W_train_weaklySup_SPS_3d_soft.py b/code/train/W_train_weaklySup_SPS_3d_soft.py
@@ -49,7 +49,7 @@
                     default='unet_cct_dp_3D', help='select mode: unet_cct_dp_3D, \
                         attention_unet_2dual_3d, unetr_2dual_3d')
 parser.add_argument('--exp', type=str,
-                    default='W_weakly_PLS_soft_3d', help='experiment_name')
+                    default='W_weakly_SPS_3d', help='experiment_name')
 parser.add_argument('--fold', type=str,
                     default='stage1', help='train fold name')
 parser.add_argument('--sup_type', type=str,
