提交 4e1d65ec 编写于 作者: xiebaiyuan's avatar xiebaiyuan

update script for mdl mobilenet convert

上级 d8d85507
......@@ -78,6 +78,10 @@ class Converter:
layers_ = self.mdl_json['layer']
for layer in layers_:
if layer['type'] == 'SoftmaxLayer':
pass
else:
desc_ops_add = block_desc.ops.add()
# print layer
......@@ -121,7 +125,7 @@ class Converter:
inputs_add = desc_ops_add.inputs.add()
inputs_add.parameter = 'X'
# todo pick last layer --> op output
inputs_add.arguments.append('Softmax')
inputs_add.arguments.append('fc7')
desc_ops_add.type = 'fetch'
outputs_add = desc_ops_add.outputs.add()
outputs_add.parameter = 'Out'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册