def setup(self): if self.exp_id == 0: # 17*17 denoising exp self.ishape = Conv2DSpace(shape=(17, 17), num_channels=1) self.path_train = "data/VOC_patch/" self.path_test = "data/test/" self.dataset_id = 2 self.batch_size = 10000 self.dl_id = str(self.exp_id) + "_" + str(self.model_id) + "_" + str(self.num_epoch) self.param_pkl = "dl_p" + self.dl_id + ".pkl" self.result_mat = "result/" + self.dl_id + "/dl_r" + str(self.test_id) + ".mat" self.param = paramSet() self.buildMLP()
def __init__(self,algo_id,model_id,num_epoch,num_dim,train_id,test_id): self.algo_id = algo_id self.model_id = model_id self.num_epoch = num_epoch self.num_dim = num_dim self.train_id = train_id self.test_id = test_id self.path_train = None self.path_test = None self.p_data = None self.batch_size = None self.do_savew = True self.param = paramSet() self.p_monitor = {}