JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
280 stars 38 forks source link

Would you please public the code for make the Brats data into the slices you mentioned in the paper? #47

Closed cqlouis closed 3 weeks ago

cqlouis commented 1 year ago

The original Brtas2020 is 155240240, but the paper said it should be change into slices with 5240240, Would you provide the code for that? Thanks advanced!

JuliaWolleb commented 1 year ago

Hi We sliced the 3D volume into 2D axial slices, and cropped them to a size of 224x224 (to cut some zero values around the brain). We stack the input to 5 channels ( 4 for the different MR sequences, and 1 for the noisy segmentation mask).