def avgpoolLayer(self, size, stride=None, pad='SAME'): self.result = L.avgpooling(self.result, size, stride, 'avgpool_' + str(self.layernum), pad=pad) self.inpsize = self.result.get_shape().as_list() return self.result
def avgpoolLayer(self, size, pad='SAME', stride=None): if stride == None: stride = size self.result = L.avgpooling(self.result, size, stride, 'maxpool_' + str(self.layernum), pad=pad) self.inpsize = self.result.get_shape().as_list() return [self.result, list(self.inpsize)]
def avgpoolLayer(self, size, pad='SAME', stride=None): if stride == None: stride = size self.result = L.avgpooling(self.result, size, stride, 'maxpool_' + str(self.layernum), pad=pad) if pad == 'VALID': self.inpsize[1] -= size - stride self.inpsize[2] -= size - stride self.inpsize[1] = self.inpsize[1] // stride self.inpsize[2] = self.inpsize[2] // stride return [self.result, list(self.inpsize)]