コード例 #1
0
ファイル: shownet.py プロジェクト: mukhalad/cuda_convnet_plus
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     self.summary_dir = self.save_file + '_summary'
     if  not os.path.exists(self.summary_dir):
         os.makedirs(self.summary_dir)
     ''' here, (train,test,pred) corresponds to (train,validation,testing)'''
     self.write_features = [self.write_features_train, self.write_features_test, self.write_features_pred]
コード例 #2
0
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     self.summary_dir = self.save_file + '_summary'
     if not os.path.exists(self.summary_dir):
         os.makedirs(self.summary_dir)
     ''' here, (train,test,pred) corresponds to (train,validation,testing)'''
     self.write_features = [
         self.write_features_train, self.write_features_test,
         self.write_features_pred
     ]
コード例 #3
0
ファイル: predict.py プロジェクト: Cysu/cuda-convnet
    def __init__(self, model_path, data_processor, gpu, layers):
        op = ConvNetPredict.get_options_parser()
        op.set_value('load_file', model_path)
        op.set_value('gpu', str(gpu))

        load_dic = IGPUModel.load_checkpoint(model_path)
        old_op = load_dic["op"]
        old_op.merge_from(op)
        op = old_op
        op.eval_expr_defaults()

        ConvNet.__init__(self, op, load_dic)

        self.dp = data_processor
        self.ftr_layer_idx = map(self.get_layer_idx, layers)
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
コード例 #5
0
ファイル: shownet.py プロジェクト: bogste/kaggle-cifar
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     self.op.set_value('data_path', '/home/komet/software/cuda-convnet/data/cifar-10-py-colmajor')
コード例 #6
0
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     self.op.set_value(
         'data_path',
         '/home/komet/software/cuda-convnet/data/cifar-10-py-colmajor')
コード例 #7
0
ファイル: accvdemo.py プロジェクト: chen1474147/convnet
 def __init__(self, op, load_dict):
     ConvNet.__init__(self, op, load_dic)
     self.statistics = dict()
     self.temp_data = dict()
コード例 #8
0
ファイル: shownet.py プロジェクト: deniskolokol/dlic
 def __init__(self, op, load_dic, dp_params):
     ConvNet.__init__(self, op, load_dic, dp_params)
コード例 #9
0
 def __init__(self, op, load_dict):
     ConvNet.__init__(self, op, load_dic)
     self.statistics = dict()
     self.temp_data = dict()
     self.cropper_dict = {'faceubd': IFaceUbdCropImage}
コード例 #10
0
ファイル: shownet.py プロジェクト: caomw/cuda-convnet2-1
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     print self.save_figure
コード例 #11
0
ファイル: accvdemo.py プロジェクト: chen1474147/convnet
 def __init__(self, op, load_dict):
     ConvNet.__init__(self, op, load_dic)
     self.statistics = dict()
     self.temp_data = dict()
コード例 #12
0
 def __init__(self, op, load_dic):
     ConvNet.__init__(self, op, load_dic)
     print self.save_figure
コード例 #13
0
 def __init__(self, op, load_dic, dp_params={}):
     filename_options = []
     self.regular_test_outputs = []
     ConvNet.__init__(self, op, load_dic, dp_params=dp_params)
     self.model_state['regular_test_outputs'] = self.regular_test_outputs
コード例 #14
0
 def __init__(self, op, load_dict):
     ConvNet.__init__(self, op, load_dic)
     self.statistics = dict()
     self.temp_data = dict()
     self.cropper_dict = {'faceubd':IFaceUbdCropImage}