PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting
MIT License
6.84k stars 953 forks source link

The link for the trained model is broken #204

Closed ohkwon718 closed 1 year ago

ohkwon718 commented 1 year ago

image

Hi. The Google Drive link seems broken. Can you fix the link or re-upload the weight? Thank you.

PeterL1n commented 1 year ago

I can't find a backup for the weights... Anyone else has the weights please share it back.

ohkwon718 commented 1 year ago

Do you think someone can manage to train the model and reproduce a similar result by himself? According to the description, the training seems to require "train_base.pth" and "train_refine.pth" files.

"Configure data_path.pth to point to your dataset. The original paper uses train_base.pth to train only the base model till convergence then use train_refine.pth to train the entire network end-to-end. More details are specified in the paper."

PeterL1n commented 1 year ago

The link is restored :)

ohkwon718 commented 1 year ago

Thank you very much :)