search

PeterL1n / BackgroundMattingV2

Real-Time High-Resolution Background Matting
MIT License
6.81k stars 950 forks source link

Pretrained model of base network #170

Closed linjing7 closed 2 years ago

linjing7 commented 2 years ago

Hi, thank you very much for your excellent work. I wonder whether the pretrained model of the base network is available? (That is, the pretrained model to be loaded in "train base.py" script.) If not, could you please provide it?

PeterL1n commented 2 years ago

It's the same weight as the final released script. You can just delete the refiner from state_dict.