darsia.wasserstein_distance

wasserstein_distance(mass_src, mass_dst, method='newton', weight=None, **kwargs)

Unified access to Wasserstein distance computation between images with same mass.

Parameters:

mass_srcdarsia.Image

Source distribution.

mass_dstdarsia.Image

Destination distribution.

methodLiteral[“newton”, “bregman”, “gprox”, “cv2.emd”]

Method to use.

**kwargs

Additional arguments (only for “newton”, “bregman”, “gprox”).

• options (dict): options for the method.

Returns:

float | tuple[float, dict

Wasserstein distance or (distance, info).