Code used for WACV submission