Focal stack images for training and inference

[SuteraBlu]

Correct me if I am wrong:

1. During inference, we need to prepare focal stack of 5 images for each scene.
2. During training we need only one image with corresponding depth map, no focal stack needed.

Did I understand the model correctly?