| 203 | |
| 204 | template <typename Dtype> |
| 205 | void SPPLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top, |
| 206 | const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) { |
| 207 | if (!propagate_down[0]) { |
| 208 | return; |
| 209 | } |
| 210 | if (pyramid_height_ == 1) { |
| 211 | pooling_layers_[0]->Backward(top, propagate_down, bottom); |
| 212 | return; |
| 213 | } |
| 214 | vector<bool> concat_propagate_down(pyramid_height_, true); |
| 215 | concat_layer_->Backward(top, concat_propagate_down, concat_bottom_vec_); |
| 216 | for (int i = 0; i < pyramid_height_; i++) { |
| 217 | flatten_layers_[i]->Backward( |
| 218 | *flatten_top_vecs_[i], propagate_down, *pooling_top_vecs_[i]); |
| 219 | pooling_layers_[i]->Backward( |
| 220 | *pooling_top_vecs_[i], propagate_down, *pooling_bottom_vecs_[i]); |
| 221 | } |
| 222 | split_layer_->Backward(split_top_vec_, propagate_down, bottom); |
| 223 | } |
| 224 | |
| 225 | INSTANTIATE_CLASS(SPPLayer); |
| 226 | REGISTER_LAYER_CLASS(SPP); |