AWA_BoundaryDoULoss

🔥 Our paper has been accepted at Neural Networks !

Introduction

This repository is an implementation of the paper "Adaptive Window Adjustment with Boundary DoU Loss for Cascade Segmentation of Anatomy and Lesions in Prostate Cancer using bpMRI".

1. Cascade Segmentation

2. Adaptive Window Adjustment Module

3. Boundary DoU Loss

Contents

• Adaptive Window Adjustment (AWA)
• BoundaryDoULoss
• Experiments on the Prostate158+ProstateX dataset
• Experiments on the PI-CAI dataset
• Experiments on the MSD dataset
• manuscript data

Datasets

• PI-CAI_image
• PI-CAI_label
• Prostate158_train_data
• Prostate158_test_data
• ProstateX
• Medical Segmentation Decathlon

Requirements

pip install -r requirements.txt

Acknowledgements

• https://github.com/sunfan-bvb/BoundaryDoULoss
• https://github.com/MGH-LMIC/windows_optimization
• https://github.com/kbressem/prostate158
• https://github.com/rcuocolo/PROSTATEx_masks
• https://github.com/Yukiya-Umimi/ITUNet-for-PICAI-2022-Challenge