You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
n, w, h, c=sizeifsize[-1]==3elsetuple(size[i] foriin [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
The text was updated successfully, but these errors were encountered:
vortex/src/runtime/vortex/runtime/onnx/onnxruntime.py
Line 118 in 9eadc6f
size is in nwhc format, assume
size=[1,32,32,3]
then n,w,h,c is correctsize is in nchw format, assume
size=[1,3,32,32]
thenn=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
The text was updated successfully, but these errors were encountered: