/Building-pose-estimation-from-the-perspective-of-UAVs-based-on-CNNs

Code on Building pose estimation from the perspective of UAVs based on CNNs

Primary LanguagePython

Building-pose-estimation-from-the-perspective-of-UAVs-based-on-CNNs

This is the code on Building pose estimation from the perspective of UAVs based on CNNs.

Tested environment

  • Ubuntu 20.04
  • Python 3.6
  • Pytorch 1.4.0
  • CUDA 10.1

Dataset

You can download our BD3D dataset at the following link.BD3D-(ef5g)

Training and Test

python Train.py
python Eval.py