1 from __future__
import absolute_import
2 from __future__
import division
3 from __future__
import print_function
4 from __future__
import unicode_literals
6 from caffe2.proto
import caffe2_pb2
9 def _parseFile(filename):
10 out_net = caffe2_pb2.NetDef()
12 dir_path = os.path.dirname(__file__)
13 with open(
'{dir_path}/{filename}'.format(dir_path=dir_path,
14 filename=filename),
'rb')
as f:
15 out_net.ParseFromString(f.read())
19 init_net = _parseFile(
'init_net.pb')
20 predict_net = _parseFile(
'predict_net.pb')