Skip to content

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

Notifications You must be signed in to change notification settings

DSaurus/LocalTrans

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published