Segmentation accuracy and network architecture in the code.

Segmentation accuracy and network architecture in the code. #364