Skip to content

Commit 5c900f7

Browse files
committed
refine ssd detection for single branch case
1 parent 6566bc7 commit 5c900f7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/caffe/layers/detection_output_layer.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ void DetectionOutputLayer<Dtype>::LayerSetUp(
147147
conf_concat_ = false;
148148
loc_concat_ = true;
149149
priorbox_concat_ = true;
150-
} else if (bottom.size() >= 2 * nbottom_ + 1 &&
151-
bottom.size() < 3 * nbottom_) {
150+
} else if ((bottom.size() >= 2 * nbottom_ + 1 &&
151+
bottom.size() < 3 * nbottom_) || nbottom_ == 1 ) {
152152
conf_concat_ = false;
153153
loc_concat_ = false;
154154
priorbox_concat_ = true;

0 commit comments

Comments
 (0)