AssertionError

[otsebriy]

AssertionError Traceback (most recent call last)
in
----> 1 model = wdsr_b(scale=scale, num_res_blocks=depth)
2 model.load_weights(weights_file)

~/Desktop/GAN-research/super-resolution/model/wdsr.py in wdsr_b(scale, num_filters, num_res_blocks, res_block_expansion, res_block_scaling)
12
13 def wdsr_b(scale, num_filters=32, num_res_blocks=8, res_block_expansion=6, res_block_scaling=None):
---> 14 return wdsr(scale, num_filters, num_res_blocks, res_block_expansion, res_block_scaling, res_block_b)
15
16

~/Desktop/GAN-research/super-resolution/model/wdsr.py in wdsr(scale, num_filters, num_res_blocks, res_block_expansion, res_block_scaling, res_block)
20
21 # main branch
---> 22 m = conv2d_weightnorm(num_filters, 3, padding='same')(x)
23 for i in range(num_res_blocks):
24 m = res_block(m, num_filters, res_block_expansion, kernel_size=3, scaling=res_block_scaling)

~/Desktop/GAN-research/super-resolution/model/wdsr.py in conv2d_weightnorm(filters, kernel_size, padding, activation, **kwargs)
57
58 def conv2d_weightnorm(filters, kernel_size, padding='same', activation=None, **kwargs):
---> 59 return tfa.layers.WeightNormalization(Conv2D(filters, kernel_size, padding=padding, activation=activation, **kwargs), data_init=False)

~/.pyenv/versions/3.6.8/envs/srgan/lib/python3.6/site-packages/typeguard/init.py in wrapper(*args, **kwargs)
1030 memo = _CallMemo(python_func, _localns, args=args, kwargs=kwargs)
1031 check_argument_types(memo)
-> 1032 retval = func(*args, **kwargs)
1033 try:
1034 check_return_type(retval, memo)

~/.pyenv/versions/3.6.8/envs/srgan/lib/python3.6/site-packages/tensorflow_addons/layers/wrappers.py in init(self, layer, data_init, **kwargs)
58 @TypeChecked
59 def init(self, layer: tf.keras.layers, data_init: bool = True, **kwargs):
---> 60 super().init(layer, **kwargs)
61 self.data_init = data_init
62 self._track_trackable(layer, name="layer")

~/.pyenv/versions/3.6.8/envs/srgan/lib/python3.6/site-packages/keras/layers/wrappers.py in init(self, layer, **kwargs)
44
45 def init(self, layer, **kwargs):
---> 46 assert isinstance(layer, Layer)
47 self.layer = layer
48 super(Wrapper, self).init(**kwargs)

AssertionError:

I wasn't able to download weights for WDSR, any ideas what's the problem?

[adavradou]

Hello @otsebriy! Same error (AssertionError) here, did you manage to find a solution?
I downloaded the WDSR weights though with no problem.

[adavradou]

Hi! I think I managed to solve the AssertionError.

According to this https://www.reddit.com/r/tensorflow/comments/dn0hjv/applying_weight_normalization_layer_in_tf_2/, you need to modify the conv2d_weightnorm function in the following way :

def conv2d_weightnorm(filters, kernel_size, padding='same', activation=None, **kwargs): return Conv2D(filters, kernel_size, padding=padding, activation=activation, **kwargs)

For me it worked!

[adavradou]

Now I get a ValueError: axes don't match array error when loading the weights though...