Exemplo n.º 1
0
 def build_dataset_test(self):
     """
     Create testing dataset
     """
     self.dataset_test = dataset.SURREAL(train=False)
     self.dataloader_test = torch.utils.data.DataLoader(self.dataset_test, batch_size=5,
                                                        shuffle=False, num_workers=int(self.opt.workers),
                                                        drop_last=True)
     self.len_dataset_test = len(self.dataset_test)
Exemplo n.º 2
0
 def build_dataset_train(self):
     """
     Create training dataset
     """
     self.dataset_train = dataset.SURREAL(train=True, regular_sampling=True)
     self.dataloader_train = torch.utils.data.DataLoader(self.dataset_train, batch_size=self.opt.batch_size,
                                                         shuffle=True, num_workers=int(self.opt.workers),
                                                         drop_last=True)
     self.len_dataset = len(self.dataset_train)