Fix: mean shape in compatible with input shape

This commit is to fix issue BVLC#5718. * reference: 1. https://groups.google.com/forum/#!topic/caffe-users/nBpWJCcJoCU 2. https://stackoverflow.com/questions/28692209/using-gpu-despite-setting-cpu-only-yielding-unexpected-keyword-argument Signed-off-by: Geunsik Lim <[email protected]>
dolfly · Nov 2, 2017 · bfc638d · bfc638d
1 parent 4efdf7e
commit bfc638d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/caffe/io.py b/python/caffe/io.py
@@ -256,7 +256,12 @@ def set_mean(self, in_, mean):
             if len(ms) != 3:
                 raise ValueError('Mean shape invalid')
             if ms != self.inputs[in_][1:]:
-                raise ValueError('Mean shape incompatible with input shape.')
+                in_shape = self.inputs[in_][1:]
+                m_min, m_max = mean.min(), mean.max()
+                normal_mean = (mean - m_min) / (m_max - m_min)
+                mean = resize_image(normal_mean.transpose((1,2,0)),
+                        in_shape[1:]).transpose((2,0,1)) * \
+                        (m_max - m_min) + m_min
         self.mean[in_] = mean
 
     def set_input_scale(self, in_, scale):