Map dimension: torch.Size([226, 185, 169])
Preprocessing time: 3.11 seconds
UNet Inference time: 9.05 seconds
Saving outputs time: 0.06 seconds
Post-processing time: 27.94 seconds
Total time: 40.16 seconds
