liziwenzzzz/pytorch-differentiable-histogram

PyTorch implementation for differentiable histogram

pytorch-differentiable-histogram

DEMO

>> python demo.py

Reference

[1] valerystrizh/pytorch-histogram-loss [link]

[2] "Learning Deep Embeddings with Histogram Loss", NeurIPS 2016 [link]