JuliaWolleb / Diffusion-based-Segmentation

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

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

Open cqlouis opened 9 months ago

cqlouis commented 9 months 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 8 months 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).