Replies: 1 comment 2 replies
-
Hi @Yangtting , Thanks for your interest here. Thanks. |
Beta Was this translation helpful? Give feedback.
2 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I used "spleen_segmentation_3d.ipynb"on my dataset.
My dataset included images and labels (0:background, 1:aortic root, 2:Plaque)
I changed model as
model = UNet(
dimensions=3,
in_channels=1,
out_channels=3,
channels=(16, 32, 64, 128, 256),
strides=(2, 2, 2, 2),
num_res_units=2,
norm=Norm.BATCH,
).to(device)
And change
"
post_pred = Compose([EnsureType(), AsDiscrete(argmax=True, to_onehot=3)])
post_label = Compose([EnsureType(), AsDiscrete(to_onehot=3)])
"
When I run code, it occured bug in valiadation process.
"AssertionError: One of self.n_classes or n_classes must be an integer"
"RuntimeError: applying transform <monai.transforms.post.array.AsDiscrete object at 0x000001EE6C51EF70>"
Beta Was this translation helpful? Give feedback.
All reactions