diff --git a/x2paddle/op_mapper/caffe_op_mapper.py b/x2paddle/op_mapper/caffe_op_mapper.py index abdbbaef11a45af55657a15d1c2b2536383df8f5..e398f2b120d306733759d10eb0a90eb3da7c7e86 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)