mirror of
https://github.com/wassname/PSPNet-Keras-tensorflow.git
synced 2026-09-12 12:01:22 +08:00
Committed caffe-tensorflow modifications to PSPNet BN layer
This commit is contained in:
@@ -104,6 +104,9 @@ class TensorFlowMapper(NodeMapper):
|
||||
def map_relu(self, node):
|
||||
return TensorFlowNode('relu')
|
||||
|
||||
def map_interp(self, node):
|
||||
return TensorFlowNode('relu')
|
||||
|
||||
def map_pooling(self, node):
|
||||
pool_type = node.parameters.pool
|
||||
if pool_type == 0:
|
||||
@@ -150,6 +153,12 @@ class TensorFlowMapper(NodeMapper):
|
||||
kwargs = {} if scale_offset else {'scale_offset': False}
|
||||
return MaybeActivated(node, default=False)('batch_normalization', **kwargs)
|
||||
|
||||
def map_bn(self, node):
|
||||
scale_offset = len(node.data) == 4
|
||||
print node
|
||||
kwargs = {} if scale_offset else {'scale_offset': False}
|
||||
return MaybeActivated(node, default=False)('batch_normalization', **kwargs)
|
||||
|
||||
def map_eltwise(self, node):
|
||||
operations = {0: 'multiply', 1: 'add', 2: 'max'}
|
||||
op_code = node.parameters.operation
|
||||
|
||||
Reference in New Issue
Block a user