LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation

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“.

Requirements

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

Pretrained Model

We have provided pretrained model under several settings in One Drive

Training

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

sh train.sh

Testing

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

sh test.sh

Citation

@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},
}