文件名称：unbalanced_gromov_wasserstein:Gromov-Wasserstein距离到不平衡最优运输设置的实现
文件大小：17KB
文件格式：ZIP
更新时间：2024-04-26 15:14:45
Python
不平衡的Gromov-Wasserstein散度 该存储库包含在》一文中提出 -Wasserstein发散的pytorch上的实现。 它允许比较配备成本矩阵的加权点云。 如果您将这项工作用于研究，请引用以下文章： @article{sejourne2020unbalanced, title={The Unbalanced Gromov Wasserstein Distance: Conic Formulation and Relaxation}, author={S{\'e}journ{\'e}, Thibault and Vialard, Fran{\c{c}}ois-Xavier and Peyr{\'e}, Gabriel}, journal={arXiv preprint arXiv:2009.04266}, year={2020} }
【文件预览】：
unbalanced_gromov_wasserstein-master
----examples()
--------plot_matching_outlier.py(4KB)
--------plot_matching_imbalanced_classes.py(4KB)
--------plot_planar_graph_match.py(11KB)
----LICENSE(1KB)
----README.md(705B)
----solver()
--------utils_numpy.py(165B)
--------__init__.py(358B)
--------utils_pytorch.py(2KB)
--------vanilla_sinkhorn_solver.py(12KB)
--------batch_sinkhorn_solver.py(12KB)
--------tlb_kl_sinkhorn_solver.py(7KB)