def testSimple(self):
net = caffe2_pb2.NetDef()
meta = metanet_pb2.MetaNetDef()
# if metanet_pb2 wasn't initialized properly the following fails with a
# cryptic message: "Parameter to MergeFrom() must be instance of same
# class: expected caffe2.NetDef got caffe2.NetDef."
meta.nets.add(key="foo", value=net)
def test_adding_net(self):
'''
Tests that NetDefs can be added to MetaNetDefs
'''
meta_net_def = metanet_pb2.MetaNetDef()
net_def = caffe2_pb2.NetDef()
meta_net_def.nets.add(key="test_key", value=net_def)
def test_replace_blobs(self):
'''
Tests that NetDefs can be added to MetaNetDefs
'''
meta_net_def = metanet_pb2.MetaNetDef()
blob_name = "Test"
blob_def = ["AA"]
blob_def2 = ["BB"]
replaced_blob_def = ["CC"]
pred_utils.AddBlobs(meta_net_def, blob_name, blob_def)
self.assertEqual(blob_def, pred_utils.GetBlobs(meta_net_def,
blob_name))
pred_utils.AddBlobs(meta_net_def, blob_name, blob_def2)
self.assertEqual(blob_def + blob_def2,
pred_utils.GetBlobs(meta_net_def, blob_name))
pred_utils.ReplaceBlobs(meta_net_def, blob_name, replaced_blob_def)
self.assertEqual(replaced_blob_def,
pred_utils.GetBlobs(meta_net_def, blob_name))
def get_meta_net_def(predictor_export_meta, ws=None, db_type=None):
"""
"""
ws = ws or workspace.C.Workspace.current
meta_net_def = metanet_pb2.MetaNetDef()
# Predict net is the core network that we use.
utils.AddNet(meta_net_def, predictor_export_meta.predict_init_name(),
utils.create_predict_init_net(ws, predictor_export_meta))
utils.AddNet(meta_net_def, predictor_export_meta.global_init_name(),
_global_init_net(predictor_export_meta, db_type))
utils.AddNet(meta_net_def, predictor_export_meta.predict_net_name(),
utils.create_predict_net(predictor_export_meta))
utils.AddBlobs(meta_net_def, predictor_export_meta.parameters_name(),
predictor_export_meta.parameters)
utils.AddBlobs(meta_net_def, predictor_export_meta.inputs_name(),
predictor_export_meta.inputs)
utils.AddBlobs(meta_net_def, predictor_export_meta.outputs_name(),
predictor_export_meta.outputs)
return meta_net_def
