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