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)