2d3d train_unet2.py is currently used for training the models. cab1 and cab2 in /segmentation_models are proposed architecture 1 and 2 respectively. TODO: Organize files Rename models to more appropriate names