Map dimension: torch.Size([154, 189, 163])
Preprocessing time: 0.51 seconds
UNet Inference time: 7.13 seconds
Saving outputs time: 0.05 seconds
Post-processing time: 2.58 seconds
Total time: 10.27 seconds
