示例#1
0
文件: P_dn_bk.py 项目: caomw/Deep_low
 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()
示例#2
0
    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 = {}