Deep-Flow-Guided-Video-Inpainting
Deep-Flow-Guided-Video-Inpainting copied to clipboard
stuck in `Loading ./pretrained_models/FlowNet2_checkpoint.pth.tar`
hello, when I run python tools/video_inpaint.py --frame_dir ./demo/frames --MASK_ROOT ./demo/masks --img_size 512 832 --FlowNet2 --DFC --ResNet101 --Propagation
, it still is stuck in
This Video has 80 Images
The optical flow list has been generated: ./demo/video_flow.txt
====> Loading ./pretrained_models/FlowNet2_checkpoint.pth.tar
anyone can give some advise? @nbei @metalbubble
Could you tell me more information about your system and GPUs?
@nbei V100. Now it works when I install correct version of torch0.4.0 by pip install https://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl
as I use cuda9.0.
Still thanks.
Can you send me FlowNet2_checkpoint.pth.tar? Thank you.
@IvyGongoogle where do you get the FlowNet2_checkpoint.pth.tar?