search

dluvizon / deephar

Deep human action recognition and pose estimation
MIT License
385 stars 83 forks source link

TypeError: lin_interpolation_2d() got an unexpected keyword argument 'dim' #22

Closed kmamine closed 4 years ago

kmamine commented 4 years ago

Hi , I'm trying to use the build_softargmax_2d block in my own network,

but when compiling the model i get the following error

`---------------------------------------------------------------------------

TypeError Traceback (most recent call last)

in () ----> 1 model = get_model() 2 model.summary() 1 frames in get_model() 19 num_rows, num_cols, num_filters = K.int_shape(x)[1:] 20 sams_input_shape = (num_rows, num_cols, num_filters) ---> 21 sam_model = build_softargmax_2d(sams_input_shape, rho=0, name='sSAM') 22 sam_model.summary() 23 output = sam_model(x) /content/deephar/deephar/models/blocks.py in build_softargmax_2d(input_shape, rho, name) 316 x = kl_divergence_regularizer(x, rho=rho) 317 --> 318 x_x = lin_interpolation_2d(x, dim=0) 319 x_y = lin_interpolation_2d(x, dim=1) 320 x = concatenate([x_x, x_y]) TypeError: lin_interpolation_2d() got an unexpected keyword argument 'dim'` I traced the error like this blocks.py --> build_softargmax_2d(....) --> x_x = lin_interpolation_2d(x, dim=0) [line 318] while in layers.py -->lin_interpolation_2d(x, axis, vmin=0., vmax=1., name=None) [the function signature ]. How to fix this ?
PhenomenalOnee commented 4 years ago

Replace "dim" with "axis"

patrickESM commented 4 years ago

thank you @PhenomenalOnee

can confirm that this works!

dluvizon commented 4 years ago

This was also fixed in the source code. Thanks @PhenomenalOnee !