【ECCV2024】GeoText-1652 Benchmark

📚 About GeoText-1652

GeoText-1652 is a groundbreaking benchmark dataset for ECCV 2024, focusing on natural language-guided drone navigation with spatial relation matching. This dataset bridges the gap between natural language processing, computer vision, and robotics, paving the way for more intuitive and flexible drone control systems.

🌟 Key Features

• Multi-platform imagery: drone, satellite, and ground cameras
• Covers multiple universities with no overlap between train and test sets
• Rich annotations including global descriptions, bounding boxes, and spatial relations

📊 Dataset Statistics

Training and test sets all include the image, global description, bbox-text pair and building numbers. We note that there is no overlap between the 33 universities of the training set and the 39 universities of the test sets. Three platforms are considered, i.e., drone, satellite, and ground cameras.

💾 Download Links

• Google Drive: GeoText-1652 Dataset
• HuggingFace Hub:
  • Dataset: truemanv5666/GeoText1652_Dataset
  • Model: truemanv5666/GeoText1652_model

📁 Dataset Structure

This dataset is designed to support the development and testing of models in geographical location recognition, providing images from multiple views at numerous unique locations.

Directory Structure

GeoText_Dataset_Official/
├── test/
│ ├── gallery_no_train(250)/
│ │ ├── 0001/
│ │ │ ├── drone_view.jpg
│ │ │ ├── street_view.jpg
│ │ │ ├── satellite_view.jpg
│ │ ├── 0002/
│ │ ├── ... // More locations
│ │ ├── 0250/
│ ├── query(701)/
│ │ ├── 0001/
│ │ │ ├── drone_view.jpg
│ │ │ ├── street_view.jpg
│ │ │ ├── satellite_view.jpg
│ │ ├── 0002/
│ │ ├── ... // More locations
│ │ ├── 0701/
├── train/
│ ├── 0001/
│ │ ├── drone_view.jpg
│ │ ├── street_view.jpg
│ │ ├── satellite_view.jpg
│ ├── 0002/
│ ├── ... // More locations
│ ├── 0701/
├── test_951_version.json
├── train.json

Annotation Details

Example entry in train.json:

{
  "image_id": "0839/image-43.jpeg",
  "image": "train/0839/image-43.jpeg",
  "caption": "In the center of the image is a large, modern office building...",
  "sentences": [
    "The object in the center of the image is a large office building with several floors and a white facade",
    "On the upper middle side of the building, there is a street with cars driving on it",
    "On the middle right side of the building, there is a small parking lot with several cars parked in it"
  ],
  "bboxes": [
    [0.408688485622406, 0.6883664131164551, 0.38859522342681885, 0.6234817504882812],
    [0.2420489490032196, 0.3855597972869873, 0.30488067865371704, 0.2891976535320282],
    [0.7388443350791931, 0.8320053219795227, 0.5213109254837036, 0.33447015285491943]
  ]
}

• Caption: Provides a global description for the entire image.
• Sentences: Each sentence is aligned with a specific part of the image, related to the bounding boxes.
• Bounding Boxes: Specified as arrays of coordinates in the format [cx, cy, w, h].

🛠️ Setup and Usage Guide

Prerequisites

• Git
• Git Large File Storage (LFS)
• Conda

Installation Steps

1. Clone the repository:

   git clone https://github.com/MultimodalGeo/GeoText-1652.git

2. Install Miniconda:

   wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
   sh Miniconda3-latest-Linux-x86_64.sh

3. Create and activate conda environment:

   conda create -n gt python=3.8
   conda activate gt

4. Install requirements:

   cd GeoText-1652
   pip install -r requirements.txt

5. Install and configure Git LFS:

   apt install git-lfs
   git lfs install

6. Download dataset and model:

   git clone https://huggingface.co/datasets/truemanv5666/GeoText1652_Dataset
   git clone https://huggingface.co/truemanv5666/GeoText1652_model

7. Extract dataset images:

   cd GeoText1652_Dataset/images
   find . -type f -name "*.tar.gz" -print0 | xargs -0 -I {} bash -c 'tar -xzf "{}" -C "$(dirname "{}")" && rm "{}"'

8. Update configuration files:

   • Update re_bbox.yaml with correct paths
   • Update method/configs/config_swinB_384.json with correct ckpt path

Running the Model

Navigate to the method directory:

cd method

For evaluation:

python3 run.py --task "re_bbox" --dist "l4" --evaluate --output_dir "output/eva" --checkpoint "/root/GeoText-1652/GeoText1652_model/geotext_official_checkpoint.pth"

For training:

nohup python3 run.py --task "re_bbox" --dist "l4" --output_dir "output/train" --checkpoint "/root/GeoText-1652/GeoText1652_model/geotext_official_checkpoint.pth" &

📄 Citation

If you find GeoText-1652 useful for your work, please cite:

@inproceedings{chu2024towards, 
  title={Towards Natural Language-Guided Drones: GeoText-1652 Benchmark with Spatial Relation Matching}, 
  author={Chu, Meng and Zheng, Zhedong and Ji, Wei and Wang, Tingyu and Chua, Tat-Seng}, 
  booktitle={ECCV}, 
  year={2024} 
}

🙏 Acknowledgements

We would like to express our gratitude to the creators of X-VLM for their excellent work, which has significantly contributed to this project.