diff options
| author | aboisbunon <aboisbunon@users.noreply.github.com> | 2018-10-22 14:40:51 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-10-22 14:40:51 +0200 |
| commit | f9995cf190b908933d6ababa2eb2bbf46431f997 (patch) | |
| tree | d44096b1c9b857b38fedeb99f21c2e7f1049058b /examples | |
| parent | d4847a8c52c09319d751e2c7d29617398b08bae7 (diff) | |
Corrected the Sinkhorn prediction images
The prediction images for Sinkhorn used the ot_emd object instead of ot_sinkhorn
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/plot_otda_color_images.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/plot_otda_color_images.py b/examples/plot_otda_color_images.py index e77aec0..62383a2 100644 --- a/examples/plot_otda_color_images.py +++ b/examples/plot_otda_color_images.py @@ -4,7 +4,7 @@ OT for image color adaptation ============================= -This example presents a way of transferring colors between two image +This example presents a way of transferring colors between two images with Optimal Transport as introduced in [6] [6] Ferradans, S., Papadakis, N., Peyre, G., & Aujol, J. F. (2014). @@ -27,7 +27,7 @@ r = np.random.RandomState(42) def im2mat(I): - """Converts and image to matrix (one pixel per line)""" + """Converts an image to matrix (one pixel per line)""" return I.reshape((I.shape[0] * I.shape[1], I.shape[2])) @@ -115,8 +115,8 @@ ot_sinkhorn.fit(Xs=Xs, Xt=Xt) transp_Xs_emd = ot_emd.transform(Xs=X1) transp_Xt_emd = ot_emd.inverse_transform(Xt=X2) -transp_Xs_sinkhorn = ot_emd.transform(Xs=X1) -transp_Xt_sinkhorn = ot_emd.inverse_transform(Xt=X2) +transp_Xs_sinkhorn = ot_sinkhorn.transform(Xs=X1) +transp_Xt_sinkhorn = ot_sinkhorn.inverse_transform(Xt=X2) I1t = minmax(mat2im(transp_Xs_emd, I1.shape)) I2t = minmax(mat2im(transp_Xt_emd, I2.shape)) |