search

gengshan-y / VCN

Volumetric Correspondence Networks for Optical Flow, NeurIPS 2019.
https://gengshan-y.github.io/VCN/
MIT License
154 stars 25 forks source link

Visualize.ipynb #9

Open realleyriley opened 4 years ago

realleyriley commented 4 years ago

Hello, I am trying to run the example notebook. However, when I run this line flow,entropy = inference(imgL_o, imgR_o, maxdisp, fac, modelpath) I get this error:

~/Documents/GitHub/VCN/models/conv4d.py in forward(self, x)
    163 
    164     def forward(self,x):
--> 165         b,c,u,v,h,w = x.size()
    166         x = self.conv1(x.view(b,c,u,v,h*w))
    167         if self.with_bn:

ValueError: not enough values to unpack (expected 6, got 5)

This happens with both the sintel and KITTI models.

gengshan-y commented 4 years ago

Do you have another version of correlation module installed? If that's the case, you could either install our modified version or simply replace self.corr() with self.corrf() in the forward function of models/VCN.py.