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

where is the resnet101_movie.pth and davis_stage3.pth?

Open mohammaddhaji opened this issue 11 months ago • 1 comments

Where can I find the resnet101_movie.pth and davis_stage3.pth? I can't find them in google drive.

FlowNet2 Inference has been finished~!
Extracted Flow has been save in ./demo/Flow
Generate datalist for initial step
Flow Completion in First Step
Freezing BatchNorm2D.
Traceback (most recent call last):
  File "tools/video_inpaint.py", line 185, in <module>
    main()
  File "tools/video_inpaint.py", line 169, in main
    flow_completion(args)
  File "tools/video_inpaint.py", line 116, in flow_completion
    test_initial_stage(args)
  File "/home/mohammad/projects/Deep-Flow-Guided-Video-Inpainting/tools/test_scripts.py", line 96, in test_initial_stage
    resume_iter = load_ckpt(args.PRETRAINED_MODEL,
  File "/home/mohammad/projects/Deep-Flow-Guided-Video-Inpainting/utils/io.py", line 24, in load_ckpt
    ckpt_dict = torch.load(ckpt_name)
  File "/home/mohammad/anaconda3/envs/dfvi/lib/python3.8/site-packages/torch/serialization.py", line 525, in load
    with _open_file_like(f, 'rb') as opened_file:
  File "/home/mohammad/anaconda3/envs/dfvi/lib/python3.8/site-packages/torch/serialization.py", line 212, in _open_file_like
    return _open_file(name_or_buffer, mode)
  File "/home/mohammad/anaconda3/envs/dfvi/lib/python3.8/site-packages/torch/serialization.py", line 193, in __init__
    super(_open_file, self).__init__(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: './pretrained_models/resnet101_movie.pth'

mohammaddhaji avatar Mar 19 '24 11:03 mohammaddhaji