Skip to content

tteepe/EarlyBird

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Early Bird 🦅

EarlyBird: Early-Fusion for Multi-View Tracking in the Bird's Eye View

Torben Teepe, Philipp Wolters, Johannes Gilg, Fabian Herzog, Gerhard Rigoll

arxiv PWC

Usage

Getting Started

  1. Install PyTorch with CUDA support
    pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
  2. Install mmcv with CUDA support
    pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
  3. Install remaining dependencies
    pip install -r requirements.txt

Training

python main.py fit -c configs/t_fit.yml \
    -c configs/d_{multiviewx,wildtrack}.yml

Testing

python main.py test -c model_weights/config.yaml \
    --ckpt model_weights/model-epoch=35-val_loss=6.50.ckpt

Acknowledgement

Cite

If you use EarlyBird, please use the following BibTeX entry.

@InProceedings{teepe2023earlybird,
    author    = {Teepe, Torben and Wolters, Philipp and Gilg, Johannes and Herzog, Fabian and Rigoll, Gerhard},
    title     = {Early{B}ird: Early-Fusion for Multi-View Tracking in the Bird's Eye View},
    booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV) Workshops},
    month     = {January},
    year      = {2024},
    pages     = {102-111}
}