summaryrefslogtreecommitdiff
path: root/examples/plot_OTDA_mapping_color_images.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/plot_OTDA_mapping_color_images.py')
-rw-r--r--examples/plot_OTDA_mapping_color_images.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/examples/plot_OTDA_mapping_color_images.py b/examples/plot_OTDA_mapping_color_images.py
index 85c4b6b..9710461 100644
--- a/examples/plot_OTDA_mapping_color_images.py
+++ b/examples/plot_OTDA_mapping_color_images.py
@@ -48,6 +48,7 @@ def mat2im(X, shape):
"""Converts back a matrix to an image"""
return X.reshape(shape)
+
X1 = im2mat(I1)
X2 = im2mat(I2)
@@ -85,6 +86,7 @@ pl.tight_layout()
def minmax(I):
return np.clip(I, 0, 1)
+
# LP problem
da_emd = ot.da.OTDA() # init class
da_emd.fit(xs, xt) # fit distributions