You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi! I am trying to train UNet-like networks available in MONAI by extending the build_network_architecture. So far, I have tried FlexibleUNet and VNet. Both performed extremely poorly. Neither of the networks apply the activation function on their output, so that isn't the issue.
The dataset in question is the TotalSegmentator dataset. I am replicating their low-res model, i.e., the network that predicts all 104 classes on 3mm isotropic spacing. The default network that uses PlainConvEncoder performs as expected. The custom (MONAI) ones fail.
This is the implementation:
from monai.networks.nets import FlexibleUNet
from .variants.network_architecture.nnUNetTrainerNoDeepSupervision import nnUNetTrainerNoDeepSupervision
class nnUNetTrainerNoDeepSupervision_ibro(nnUNetTrainerNoDeepSupervision):
@staticmethod
def build_network_architecture(plans_manager,
dataset_json,
configuration_manager,
num_input_channels,
enable_deep_supervision=False):
out_channels = plans_manager.get_label_manager(dataset_json).num_segmentation_heads
spatial_dims = len(configuration_manager.conv_kernel_sizes[0])
model = FlexibleUNet(
in_channels=num_input_channels,
out_channels=out_channels,
spatial_dims=spatial_dims,
upsample="deconv",
backbone="efficientnet-b4"
)
return model
Any idea what may be the issue?
Thank you so much!
The text was updated successfully, but these errors were encountered:
Hello I have a question if I alredy write a file like nnUNetTrainerNodeepSupervison.py. Where should I call it. I only find:
network = nw_class() in get_network_from_plans.py
Hi! I am trying to train UNet-like networks available in MONAI by extending the
build_network_architecture
. So far, I have triedFlexibleUNet
andVNet
. Both performed extremely poorly. Neither of the networks apply the activation function on their output, so that isn't the issue.The dataset in question is the TotalSegmentator dataset. I am replicating their low-res model, i.e., the network that predicts all 104 classes on 3mm isotropic spacing. The default network that uses
PlainConvEncoder
performs as expected. The custom (MONAI) ones fail.This is the implementation:
Any idea what may be the issue?
Thank you so much!
The text was updated successfully, but these errors were encountered: