Delineation of the prostate and nearby organs at risk (OARs) is a fundamental step in prostate cancer radiation therapy planning. Such contouring is often done manually, which can be a time-consuming and highly variable process. To alleviate these issues, we propose a fully automated two-step deep learning approach to segment the prostate, bladder, rectum, seminal vesicles, and femoral heads from CT images. The first step localizes the organs of interest using a modified 3D UNet architecture that contains an axial cross-attention module. Final segmentations are then computed for each organ individually using organ-specifically optimized UNet-based models. A total of 275 CT images were used for model training and validation. When evaluated on a hold-out set of 15 image sets, the full pipeline achieved mean dice similarity coefficients (DSC) and 95% Hausdorff distances (95HD, in mm) of 0.8660.034 and 4.461.02 (prostate), 0.9570.014 and 2.910.289 (bladder), 0.8530.044 and 5.101.87 (rectum), 0.7400.117 and 6.729.46 (seminal vesicles), 0.9420.016 and 2.851.04 (left femoral head), 0.9420.018 and 3.041.37 (right femoral head).
|