In this repository, you'll find an implementation of the weighted Hausdorff Distance Loss, described here (https://arxiv.org/abs/1806.07564). A majority of the work was just porting their PyTorch implementation (https://github.com/HaipengXiong/weighted-hausdorff-loss). I figured some researchers/practitioners that are doing object detection/localization may find this useful!

pipenv install . should configure a python environment and install all necessary dependencies in the environment.

Some tests verifying basic components of the loss function have been incorporated. Run python -m pytest in the repo to execute them.

Add an example script.