Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 

README.md

Weighted Hausdorff Distance Tensorflow/Keras loss

Weighted Hausdorff Distance Loss: use it as a point cloud similarity metric based loss for keras and tf. Useful in keypoint detection.

Attention

This loss requires a huge tensor with dimensions (number_of_pixels * number_of_keypoints if I remember correctly) of float values. So high res picture with thousands of keypoints will consume A LOT of GPU memory (at least 1 GB for 512 pixels x 512 pixels x 1000 keypoints with float32 type). 1024x1024x2000 will eat 8 GB. It doesn't matter if you want to detect only several points in an image. https://arxiv.org/pdf/1806.07564.pdf

About

Weighted Hausdorff Distance Loss: use it as point cloud similarity metric based loss for keras and tf. Useful in keypoint detection.

Resources

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.