Skip to content

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

License

Notifications You must be signed in to change notification settings

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

  • generate_hnet_training_data.py generates synthetic distance matrices and association matrices.
  • Then 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.