如何以十位数加载mat数据

2024-04-26 10:31:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在运行pix2pixSE(用于语音增强和抗噪声说话人验证的条件生成对抗网络代码)。当我用mat数据训练模型时,它不能成功地加载数据

我检查了数据路径和load函数的代码,但它也不起作用

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

当我训练模型时,它在加载数据后停止(我认为它没有成功地加载数据)


Tags: 数据samplepath代码name模型selfdata