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