Ruizhi Shao*, Gaochang Wu*, Yuemei Zhou, Ying Fu, Lu Fang, Yebin Liu

This repository contains the official pytorch implementation of ”LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation“.

• pytorch
• matplotlib
• numpy
• cv2
• tensorboard
• kornia
• imageio

We have provided pretrained model under several settings in One Drive

To train localtrans on the COCO dataset in different setting, run the following code:

sh train.sh

Run the following code to test on the COCO test dataset.

sh test.sh

@inproceedings{shao2021localtrans,
title={LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation},
author={Shao, Ruizhi and Wu, Gaochang and Zhou, Yuemei and Fu, Ying and Fang, Lu and Liu, Yebin},
booktitle={IEEE Conference on Computer Vision (ICCV 2021)},
year={2021},
}