def MXNetParse(architecture_name, image_path):
from mmdnn.conversion.examples.mxnet.extractor import mxnet_extractor
from mmdnn.conversion.mxnet.mxnet_parser import MXNetParser
# download model
architecture_file, weight_file = mxnet_extractor.download(architecture_name, TestModels.cachedir)
# get original model prediction result
original_predict = mxnet_extractor.inference(architecture_name, TestModels.cachedir, image_path)
del mxnet_extractor
# original to IR
import re
if re.search('.', weight_file):
weight_file = weight_file[:-7]
prefix, epoch = weight_file.rsplit('-', 1)
model = (architecture_file, prefix, epoch, [3, 224, 224])
IR_file = TestModels.tmpdir + 'mxnet_' + architecture_name + "_converted"
parser = MXNetParser(model)
parser.run(IR_file)
del parser
del MXNetParser
return original_predict
def MXNetParse(architecture_name, image_path):
from mmdnn.conversion.examples.mxnet.extractor import mxnet_extractor
from mmdnn.conversion.mxnet.mxnet_parser import MXNetParser
# download model
architecture_file, weight_file = mxnet_extractor.download(architecture_name, TestModels.cachedir)
# get original model prediction result
original_predict = mxnet_extractor.inference(architecture_name, None, TestModels.cachedir, image_path)
del mxnet_extractor
# original to IR
import re
if re.search('.', weight_file):
weight_file = weight_file[:-7]
prefix, epoch = weight_file.rsplit('-', 1)
model = (architecture_file, prefix, epoch, [3, 224, 224])
IR_file = TestModels.tmpdir + 'mxnet_' + architecture_name + "_converted"
parser = MXNetParser(model)
parser.run(IR_file)
del parser
del MXNetParser
return original_predict
def MXNetParse(architecture_name, image_path):
# download model
architecture_file, weight_file = mxnet_extractor.download(
architecture_name, TestModels.cachedir)
# get original model prediction result
original_predict = mxnet_extractor.inference(architecture_name,
TestModels.cachedir,
image_path)
# original to IR
import re
if re.search('.', weight_file):
weight_file = weight_file[:-7]
prefix, epoch = weight_file.rsplit('-', 1)
model = (architecture_file, prefix, epoch, [3, 224, 224])
parser = MXNetParser(model)
parser.gen_IR()
parser.save_to_proto(TestModels.tmpdir + architecture_name +
"_converted.pb")
parser.save_weights(TestModels.tmpdir + architecture_name +
"_converted.npy")
del parser
return original_predict
