sczhou / ProPainter

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting
https://shangchenzhou.com/projects/ProPainter/
Other
5.45k stars 645 forks source link

Hard Coded Path to Recurrent Flow Completion Model Weight #73

Open ykwongaq opened 6 months ago

ykwongaq commented 6 months ago

Thank you for the great work on ProPainter!

I am trying to train the ProPainter using the pretrained recurrent flow completion model, however, I got a file not found error:

image

I found out the root cause is at core/trainer.py line 75:

image

where the path for flow completion model is hard coded.

Perphas it is better to configure it in the train_propainter.json

Thank you

Pokerman8 commented 6 months ago

I meet the same problem, and I want to know that if the path of the Recurrent Flow Completion Model Weight is /ProPainter-main/weights/recurrent_flow_completion.pth

ykwongaq commented 6 months ago

I meet the same problem, and I want to know that if the path of the Recurrent Flow Completion Model Weight is /ProPainter-main/weights/recurrent_flow_completion.pth

Yes, I have changed the path to what you mentioned. And it is working now.