Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Problem in _to_bc_h_w_2 ?? #6

Open
mschoeler opened this issue Jul 1, 2020 · 0 comments
Open

Problem in _to_bc_h_w_2 ?? #6

mschoeler opened this issue Jul 1, 2020 · 0 comments

Comments

@mschoeler
Copy link

In deform_conv/layers.py:39

    @staticmethod
    def _to_bc_h_w_2(x, x_shape):
        """(b, h, w, 2c) -> (b*c, h, w, 2)"""
        x = tf.transpose(x, [0, 3, 1, 2])
        x = tf.reshape(x, (-1, int(x_shape[1]), int(x_shape[2]), 2))
        return x

I think the transpose changes the array from
(b, h, w, 2c) -> (b, 2c, h, w)
So the subsequent reshape will splitt the width dimension of your array.

This seems like a bug in the code to me.

Cheers

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant