Skip to content

Commit

Permalink
adding GB option and todo
Browse files Browse the repository at this point in the history
  • Loading branch information
adalca committed Sep 14, 2020
1 parent 7b8f672 commit 0fd81fc
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion neurite/tf/layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,7 +202,7 @@ class GaussianBlur(Layer):
TODO: switch from 'level' based to 'sigma' based
"""

def __init__(self, level=None, sigma=None, **kwargs):
def __init__(self, sigma=None, level=None, **kwargs):
assert sigma is not None or level is not None, 'sigma or level must be provided'
assert not (sigma is not None and level is not None), 'only sigma or level must be provided'

Expand Down Expand Up @@ -234,6 +234,7 @@ def build(self, input_shape):

def call(self, x):
# TODO: switch to mutli-line if statement
# TODO: this hsould be cleaned up a bit, no need to loop and concat...
return x if self.sigma == 0 else tf.concat([self.conv(x[..., n]) for n in range(self.nfeat)], -1)

def compute_output_shape(self, input_shape):
Expand Down

0 comments on commit 0fd81fc

Please sign in to comment.