Fill in the structure of Segmentation model.