fix the broadcasting initializer issue for power in corner case

Author: vera121

include/caffe/layers/pow_layer.hpp

@@ -21,6 +21,8 @@ class PowLayer : public Layer<Dtype> {
  public:
   explicit PowLayer(const LayerParameter& param)
       : Layer<Dtype>(param) {}
+  virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom,
+      const vector<Blob<Dtype>*>& top);
   virtual void Reshape(const vector<Blob<Dtype>*>& bottom,
       const vector<Blob<Dtype>*>& top);
 

src/caffe/layers/pow_layer.cpp

@@ -6,6 +6,26 @@
 #include "caffe/layers/pow_layer.hpp"
 
 namespace caffe {
+template <typename Dtype>
+void PowLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom,
+      const vector<Blob<Dtype>*>& top) {
+  // Add the fake initialization to avoid special case in evconvert for the moment when
+  // the prototxt is generated, the 2nd input is parameter to be fed into caffemodel, but the caffemodel is not generated yet
+  if (bottom.size() == 1 && this->blobs_.size() == 0)
+  {
+    LOG(WARNING) << "Note: Make fake parameter initialization to avoid segment fault!";
+    // initialize the 2nd input to avoid error in reshape stage
+    this->blobs_.resize(1);
+    const vector<int>::const_iterator& shape_start =
+        bottom[0]->shape().begin();
+    const vector<int>::const_iterator& shape_end = bottom[0]->shape().end();
+    vector<int> bias_shape(shape_start, shape_end);
+    this->blobs_[0].reset(new Blob<Dtype>(bias_shape));
+
+    this->param_propagate_down_.resize(this->blobs_.size(), true);
+  }
+}
+
 template <typename Dtype>
 void PowLayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom,
       const vector<Blob<Dtype>*>& top) {