From 1c46bafe7c1fe5729379125cbba0d8719060ab7c Mon Sep 17 00:00:00 2001 From: SunAhong1993 <48579383+SunAhong1993@users.noreply.github.com> Date: Tue, 3 Dec 2019 21:02:00 +0800 Subject: [PATCH] Update caffe_op_mapper.py --- x2paddle/op_mapper/caffe_op_mapper.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/caffe_op_mapper.py b/x2paddle/op_mapper/caffe_op_mapper.py index abdbbae..0fde719 100644 --- a/x2paddle/op_mapper/caffe_op_mapper.py +++ b/x2paddle/op_mapper/caffe_op_mapper.py @@ -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) -- GitLab