search

mazurowski-lab / segmentation-guided-diffusion

[MICCAI 2024] Easy diffusion models (optionally with segmentation guidance) for medical images and beyond.
https://arxiv.org/abs/2402.05210
Other
88 stars 5 forks source link

convert_segbatch_to_multiclass #10

Closed skyqwe123 closed 3 months ago

skyqwe123 commented 3 months ago

@nickk124 thanks for your job. When I was executing the code, there was an error,

in convert_segbatch_to_multiclass segs[segs == 0] = seg[segs == 0]



IndexError: The shape of the mask [32, 3, 512, 512] at index 1 does not match the shape of the indexed tensor [32, 1, 512, 512] at index 1