如何在tensorflow中加载mat数据

Question

我正在运行pix2pixSE（用于语音增强和鲁棒性说话人验证的条件生成对抗网络代码）。当我用Mat数据训练模型时，它无法成功加载数据。

我检查了数据路径和加载函数的代码，但它也不起作用。

def main（_）：

data_path = '/mnt/hd5/dm/GAN/pix2pix/T1/white' #'.'
#data_path = '/home/bob/pix2pixSE/speech enhancement/baseline/datasets'
args.checkpoint_dir=data_path+'/checkpoint' 
args.sample_dir=data_path+'/sample'
args.test_dir=data_path+'/test'`for epoch in xrange(args.epoch):
        if self.dataset_name == 'spectrograms':
            data = glob(self.db_path+'/{}/train/*.mat'.format(self.dataset_name))
        else:
            data = glob(self.db_path+'/{}/train/*.jpg'.format(self.dataset_name))

        #np.random.shuffle(data)
        batch_idxs = min(len(data), args.train_size) // self.batch_size

当我训练模型时，它在加载数据后停止（我认为它没有成功加载数据）