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

[BUG] It seems that the size formatting is only for NWHC, NCHW is not supported #119

Open
kurnianggoro opened this issue Sep 15, 2021 · 2 comments

Comments

@kurnianggoro
Copy link

kurnianggoro commented Sep 15, 2021

n, w, h, c = size if size[-1]==3 else tuple(size[i] for i in [0,3,1,2])

n, w, h, c = size if size[-1]==3 else tuple(size[i] for i in [0,3,1,2])

size is in nwhc format, assume size=[1,32,32,3] then n,w,h,c is correct
size is in nchw format, assume size=[1,3,32,32] then
n=size[0]=1
w=size[3]=32
h=size[1]=3
c=size[2]=32

It seems that the command size[i] for i in [0,3,1,2] will produce correct result only when the input size is in NWHC format.
Thus, both of the conditions are for NWHC format

@triwahyuu
Copy link
Member

oh yeah you are right @kurnianggoro, seems like we need more tests on this

@alifahrri
Copy link
Collaborator

i thought we standardized the input to be NHWC @triwahyuu ?

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

3 participants