def load_using_pb(self): import caffepb data = caffepb.NetParameter() data.MergeFromString(open(self.data_path, 'rb').read()) pair = lambda layer: (layer.name, self.transform_data(layer)) layers = data.layers or data.layer self.parameters = [pair(layer) for layer in layers if layer.blobs]
def load(self): self.params = caffepb.NetParameter() with open(self.def_path, 'rb') as def_file: text_format.Merge(def_file.read(), self.params)