未验证 提交 1c46bafe 编写于 作者: S SunAhong1993 提交者: GitHub

Update caffe_op_mapper.py

上级 8b21d123
......@@ -940,7 +940,9 @@ class CaffeOpMapper(OpMapper):
input = self.graph.get_bottom_node(node, idx=i, copy=True)
if i == 1 and op == 'DetectionOutput':
input = self.graph.get_bottom_node(node, idx=i, copy=True)
while input is not None and input.layer_type != 'Softmax':
while input is not None
and input.layer_type != 'Softmax'
and input.layer_type != 'Sigmoid':
input = self.graph.get_bottom_node(input, idx=0, copy=True)
assert input is not None, 'This kind of DetectionOutput is not supported!'
input = self.graph.get_bottom_node(input, idx=0, copy=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册