Train a 3D CNN for CT Image Segmentation in PyTorch

Mediumauto-graded

Extend 2D convolutions to 3D for volumetric medical image segmentation on CT scans.

Solve it

Open in ColabQuestion notebookSolution

Check your answer

The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.

pip install torchleet

from torchleet import check
check("3dcnn", MedCNN, compute_dice_loss)