|
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
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]=1w=size[3]=32h=size[1]=3c=size[2]=32It 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