Update run_one_image.py : Load to supplied device

hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers

MIT License

339 stars 33 forks source link

Closed abhishek0093 closed 7 months ago

abhishek0093 commented 7 months ago

The supplied device from user arguments was never being used.