Skip to content

MaloOLIVIER/hungarian-net

Repository files navigation

Python PyTorch Lightning Config: hydra Pytest

Differentiable assignation problem resolution by deep learning method.

Requirements

pip install -r requirements.txt

Getting Started

  • python generate_hnet_training_data.py generates synthetic distance matrices and association matrices.
  • Then python run.py to train Hnet with the generated data.

Sharath Adavanne*, Archontis Politis* and Tuomas Virtanen, "Differentiable Tracking-Based Training of Deep Learning Sound Source Localizers" in the IEEE Workshop on Applications of Signal Processing to Audio and Acoustics (WASPAA 2021)

License

The repository is licensed under the TAU License.

About

Hungarian Network 🔬 — Generate synthetic data and train your deep-learning implementation of the Hungarian algorithm.

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 2

  •  
  •  

Languages