search

baowenbo / DAIN

Depth-Aware Video Frame Interpolation (CVPR 2019)
https://sites.google.com/view/wenbobao/dain
MIT License
8.23k stars 840 forks source link

ValueError: axes don't match array #110

Open Eleanor-lmn opened 4 years ago

Eleanor-lmn commented 4 years ago

When I try to interpolate the frame with the processed grayscale image, an error occurs：ValueError: axes don't match array

X0 = torch.from_numpy( np.transpose(imread(arguments_strFirst) , (2,0,1)).astype("float32")/ 255.0).type(dtype) Traceback (most recent call last): File "modify_demo_MiddleBury.py", line 121, in X0 = torch.from_numpy( np.transpose(imread(arguments_strFirst) , (2,0,1)).astype("float32")/ 255.0).type(dtype)
File "/root/anaconda3/lib/python3.7/site-packages/numpy/core/fromnumeric.py", line 598, in transpose return _wrapfunc(a, 'transpose', axes) File "/root/anaconda3/lib/python3.7/site-packages/numpy/core/fromnumeric.py", line 51, in _wrapfunc return getattr(obj, method)(*args, **kwds) ValueError: axes don't match array

fedhere commented 3 years ago

I am getting the same error, any resolution?