hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers
MIT License
371 stars 37 forks source link

Pretrained Weights #3

Closed YiftachEde closed 1 year ago

YiftachEde commented 1 year ago

Hey, in your train.md you specify there is the ability to pull the pretrained weights via a script, the script and the directory it sits under do not exist. is the pretrained weights the weights for DiNO and MAE? or for the ViTMatte after fine tuning?

thanks for the great work!

YiftachEde commented 1 year ago

I see now that there is a pretrained weight download in the readme. thanks! Yiftach