Deep learning based segmentation of brain tissue from diffusion MRI