Map dimension: torch.Size([138, 186, 138])
Preprocessing time: 0.52 seconds
UNet Inference time: 4.75 seconds
Saving outputs time: 0.03 seconds
Post-processing time: 2.66 seconds
Total time: 7.96 seconds
