model_training_utils.py 文件源码

def parse_solver(self):
        solverPath = self.solverPath;
        self.expName = os.path.split(solverPath)[-1].split('_')[0];
        self.expDir = os.path.split(solverPath)[0];
        self.solver_param = caffe_pb2.SolverParameter();
        with open(self.solverPath, 'rt') as f:
            pb2.text_format.Merge(f.read(), self.solver_param)

        allLines = [x.strip() for x in open(solverPath,'r')];

        snapPath = self.solver_param.snapshot_prefix;
        snapExp = os.path.split(snapPath)[-1];
        snapPath = os.path.split(snapPath)[0];
        sg_utils.mkdir(snapPath);
        assert( os.path.isdir(snapPath) ), '%s does not exist'%(snapPath);
        self.snapPath = snapPath;
        assert( self.snapPath == os.path.split(self.solver_param.snapshot_prefix)[0] );