Map dimension: torch.Size([381, 370, 374])
Preprocessing time: 4.09 seconds
UNet Inference time: 64.69 seconds
Saving outputs time: 0.84 seconds
Post-processing time: 11.82 seconds
Total time: 81.44 seconds
