From e77514fbcb7e8aa2d91747e95e1250820d223bae Mon Sep 17 00:00:00 2001 From: Arnur Nigmetov Date: Thu, 25 Jan 2018 10:33:41 +0100 Subject: Fix for inf-only diagrams --- geom_matching/wasserstein/include/wasserstein.h | 3 +++ 1 file changed, 3 insertions(+) (limited to 'geom_matching/wasserstein/include/wasserstein.h') diff --git a/geom_matching/wasserstein/include/wasserstein.h b/geom_matching/wasserstein/include/wasserstein.h index d843a04..b90a545 100644 --- a/geom_matching/wasserstein/include/wasserstein.h +++ b/geom_matching/wasserstein/include/wasserstein.h @@ -228,6 +228,9 @@ namespace ws throw std::runtime_error("Bad epsilon factor in Wasserstein " + std::to_string(params.epsilon_common_ratio)); } + if (A.empty() and B.empty()) + return 0.0; + RealType result; // just use Gauss-Seidel -- cgit v1.2.3