Skip to content

Latest commit

 

History

History
43 lines (34 loc) · 1.35 KB

README.md

File metadata and controls

43 lines (34 loc) · 1.35 KB
Raw

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

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

report

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

Teaser Image

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