We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6566bc7 commit 5c900f7Copy full SHA for 5c900f7
src/caffe/layers/detection_output_layer.cpp
@@ -147,8 +147,8 @@ void DetectionOutputLayer<Dtype>::LayerSetUp(
147
conf_concat_ = false;
148
loc_concat_ = true;
149
priorbox_concat_ = true;
150
- } else if (bottom.size() >= 2 * nbottom_ + 1 &&
151
- bottom.size() < 3 * nbottom_) {
+ } else if ((bottom.size() >= 2 * nbottom_ + 1 &&
+ bottom.size() < 3 * nbottom_) || nbottom_ == 1 ) {
152
153
loc_concat_ = false;
154
0 commit comments