Paper accepted to CVPR 2022