I have made some progress with this using the h5py package. Depth Segmentation using U-nets but on .mat files