PROSAGA码农传奇-深度学习-Caffe python层后缀传递实现

<div class =“post-text”itemprop =“text”>
  
    如果你愿意忽略，你的功能很简单
     <code>
 np.around
 </code>
    ：
  
  
    <a href="https://i.stack.imgur.com/0kRj0.png" rel="nofollow noreferrer">
      <img src =“https://i.stack.imgur.com/0kRj0.png”alt =“在此处输入图片说明”/>
    </A>
  
  
    对于
     <code>
 x=x_min
 </code>
     并为
     <code>
 x=x_max
 </code>
     对于所有其他的，导数为零
     <code>
 x
 </code>
     衍生品是
     <code>
 255/(x_max-x_min)
 </code>
    。
  
  
    这可以通过实现
  
   <pre>
 <code>
 def forward(self, bottom, top):
 in_ = bottom[0].data
 self.x_min = in_.min(axis=(0, 1), keepdims=True) # cache min/max for backward
 self.x_max = in_.max(axis=(0, 1), keepdims=True)
 top[0].data[...] = 255*((in_-self.x_min)/(self.x_max-self.x_min)))

def backward(self, top, propagate_down, bottom):
  in_ = bottom[0].data
  b, c = in_.shape[:2]
  diff = np.tile( 255/(self.x_max-self.x_min), (b, c, 1, 1) )
  diff[ in_ == self.x_min ] = 0
  diff[ in_ == self.x_max ] = 0
  bottom[0].diff[...] = diff * top[0].diff

</code>
 </pre>
  
    别忘了测试这个数字。这可以通过例如使用来完成
    <a href="https://github.com/BVLC/caffe/pull/5157" rel="nofollow noreferrer">
       <code>
 test_gradient_for_python_layer
 </code>
    </A>
    。
  
</DIV>