https://github.com/dsaurus/localtrans
This repository is the official implementation of LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation.
https://github.com/dsaurus/localtrans
Last synced: about 1 year ago
JSON representation
This repository is the official implementation of LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation.
- Host: GitHub
- URL: https://github.com/dsaurus/localtrans
- Owner: DSaurus
- Created: 2022-02-04T05:02:26.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2022-02-05T16:14:07.000Z (over 4 years ago)
- Last Synced: 2025-04-19T14:57:47.109Z (about 1 year ago)
- Language: C++
- Homepage:
- Size: 1.26 MB
- Stars: 23
- Watchers: 1
- Forks: 2
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# [LocalTrans: A Multiscale Local Transformer Network for Cross-Resolution Homography Estimation](http://www.liuyebin.com/localtrans/localtrans.html)
Ruizhi Shao*, Gaochang Wu*, Yuemei Zhou, Ying Fu, Lu Fang, Yebin Liu
[](https://arxiv.org/abs/2106.04067)
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](https://mailstsinghuaeducn-my.sharepoint.com/:f:/g/personal/shaorz20_mails_tsinghua_edu_cn/Et6rFUvWy8VNjC6gDWhpOSoBjZ9ISDTGkaTBumLafQ9asw?e=IQe8Or)
## 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},
}
```