Deep-Flow-Guided-Video-Inpainting icon indicating copy to clipboard operation
Deep-Flow-Guided-Video-Inpainting copied to clipboard

Published 20 hours ago • verified publisher icon nbei

about Correlation_forward_cuda

Open lvzhengyi0204 opened this issue 4 years ago • 6 comments

Hello： I have configured the environment according to your requirements and executed the command 1：bash install_scripts.sh but when I run the commant：python3 tools/video_inpaint.py --frame_dir ./demo/frames --MASK_ROOT ./demo/masks --img_size 512 832 --FlowNet2 --DFC --ResNet101 --Propagation there has a error: AttributeError: module 'models.FlowNet2_Models.correlation_package._ext.correlation' has no attribute 'Correlation_forward_cuda'

lvzhengyi0204 avatar Apr 29 '20 09:04 lvzhengyi0204

Have you checked the version of PyTorch? And here is another branch without installation of flownet2.

nbei avatar Apr 29 '20 10:04 nbei

torch 0.4.0
torchvision 0.2.1 I completely configure the environment according to your requirements

lvzhengyi0204 avatar Apr 29 '20 10:04 lvzhengyi0204

image the correlation has No this attribute:Correlation_forward_cuda

lvzhengyi0204 avatar Apr 29 '20 10:04 lvzhengyi0204

please help me~ hahaha

lvzhengyi0204 avatar Apr 29 '20 10:04 lvzhengyi0204

I suggest that you can just compile the correlation layer and then see whether any error information will be reported.

nbei avatar Apr 29 '20 11:04 nbei

I gave up flownet2 and switched to lightflow, and I can experiment normally

lvzhengyi0204 avatar Apr 29 '20 13:04 lvzhengyi0204