Paper accepted at CV4MS @ CVPR 2024