PyVot
A Python implementation of variational optimal transportation
This package includes the prototype codes for reproducing the map in Figure 4 (b) of the paper:
Mi, Liang, Wen Zhang, Xianfeng Gu, and Yalin Wang. "Variational Wasserstein Clustering." In Proceedings of the European Conference on Computer Vision (ECCV), pp. 322-337. 2018.
Dependencies
- Python >= 3.5
- NumPy >= 1.15.4
- SciPy >= 1.1.0
- Matplotlib >= 3.0.2 (for demo only)
Reference
If you find the code helpful, please cite the following article:
Mi, Liang, Wen Zhang, Xianfeng Gu, and Yalin Wang. "Variational Wasserstein Clustering." In Proceedings of the European Conference on Computer Vision (ECCV), pp. 322-337. 2018.
@inproceedings{mi2018variational,
title={Variational {W}asserstein Clustering},
author={Mi, Liang and Zhang, Wen and Gu, Xianfeng and Wang, Yalin},
booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
pages={322--337},
year={2018}
}
Contact
Please contact Liang Mi icemiliang@gmail.com for any issues.