add notebook for color transfer in images

Author: rflamary

README.md

@@ -64,6 +64,8 @@ The examples folder contain several examples and use case for the library. The f
 * [1D Wasserstein barycenter](https://github.com/rflamary/POT/blob/master/examples/Demo_1D_barycenter.ipynb)
 * [OT with user provided regularization](https://github.com/rflamary/POT/blob/master/examples/Demo_Optim_OTreg.ipynb)
 * [Domain adaptation with optimal transport](https://github.com/rflamary/POT/blob/master/examples/Demo_2D_OT_DomainAdaptation.ipynb)
+* [Color transfer in images](https://github.com/rflamary/POT/blob/master/examples/Demo_Image_ColorAdaptation.ipynb)
+
 
 ## Acknowledgements
 

docs/source/examples.rst

@@ -27,3 +27,8 @@ Domain adaptation with optimal transport
 ----------------------------------------
 
 .. literalinclude:: ../../examples/demo_OTDA_classes.py
+
+Color transfer in images
+------------------------
+
+.. literalinclude:: ../../examples/demo_OTDA_color_images.py

examples/Demo_Image_ColorAdaptation.ipynb

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """
-demo of Optimal transport for domain adaptation with image color adaptation as in [6]
+Demo of Optimal transport for domain adaptation with image color adaptation as in [6]
 
 [6] Ferradans, S., Papadakis, N., Peyré, G., & Aujol, J. F. (2014). Regularized discrete optimal transport. SIAM Journal on Imaging Sciences, 7(3), 1853-1882.
 """
@@ -51,6 +51,30 @@ def mat2im(X,shape):
 xs=X1[idx1,:]
 xt=X2[idx2,:]
 
+#%% Plot image distributions
+
+
+pl.figure(2,(10,5))
+
+pl.subplot(1,2,1)
+pl.scatter(xs[:,0],xs[:,2],c=xs)
+pl.axis([0,1,0,1])
+pl.xlabel('Red')
+pl.ylabel('Blue')
+pl.title('Image 1')
+
+pl.subplot(1,2,2)
+#pl.imshow(I2)
+pl.scatter(xt[:,0],xt[:,2],c=xt)
+pl.axis([0,1,0,1])
+pl.xlabel('Red')
+pl.ylabel('Blue')
+pl.title('Image 2')
+
+pl.show()
+
+
+
 #%% domain adaptation between images
 
 # LP problem