def load(self): self.model = get_caffe_resolver().NetParameter() with open(self.model_path, 'r') as f: text_format.Merge(f.read(), self.model) if self.is_train_proto: self.process_train_proto()