Paper accepted at ICCV 2021.