Thanks to this effort, I tried to reproduce the BRATS2021 dataset, and I modified the saved file to NII format. I wondered why the final sample result was an MRI image without a mask, how do I get the result of the generated prediction mask to compute the metric?