Home

Awesome

<h1 align="center"> FSRA </h1>

This repository contains the dataset link and the code for our paper A Transformer-Based Feature Segmentation and Region Alignment Method For UAV-View Geo-Localization, IEEE Transactions on Circuits and Systems for Video Technology. Thank you for your kindly attention.

requirement

  1. Download the University-1652 dataset
  2. Configuring the environment
    • First you need to configure the torch and torchision from the pytorch website
    • pip install -r requirement.txt
      

Download pre-training weights

You can download the pre-training weight from the following link and put them in the pretrain_model folder

  1. Baidu Driver code: 0000

  2. Google Driver

Train and Test

We provide scripts to complete FSRA training and testing

bash train_test_local.sh