update: parcellation

src/utils/parcellation.py

@@ -75,20 +75,25 @@ def parcellation(voxel, pnet_c, pnet_s, pnet_a, device):
     axial = voxel.transpose(2, 1, 0)
 
     # Perform parcellation for the coronal view
+    print("Performing parcellation for coronal view...")
     out_c = parcellate(coronal, pnet_c, device, "c").permute(1, 3, 0, 2)
     torch.cuda.empty_cache()
+    print("Parcellation for coronal view completed.")
 
     # Perform parcellation for the sagittal view
+    print("Performing parcellation for sagittal view...")
     out_s = parcellate(sagittal, pnet_s, device, "s").permute(1, 0, 2, 3)
     torch.cuda.empty_cache()
+    print("Parcellation for sagittal view completed.")
 
     # Combine the results from coronal and sagittal views
     out_e = out_c + out_s
     del out_c, out_s
-
+    print("Combining results from coronal and sagittal views...")
     # Perform parcellation for the axial view
     out_a = parcellate(axial, pnet_a, device, "a").permute(1, 3, 2, 0)
     torch.cuda.empty_cache()
+    print("Parcellation for axial view completed.")
 
     # Combine the results from all views
     out_e = out_a + out_e