diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 14:50:02 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 14:50:02 +0200 |
| commit | d5efa20c70a6988a251acc15759dcd9d182c6dc1 (patch) | |
| tree | 194dbbfa5d2a6eb88983e5a7da2bb204ed5c3031 | |
| parent | 1ea47b72fa3ee8e9ee4c9a1e17871506b4d50066 (diff) | |
read image using pylab
| -rwxr-xr-x | examples/plot_gromov_barycenter.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/examples/plot_gromov_barycenter.py b/examples/plot_gromov_barycenter.py index 58fc51a..101c6c5 100755 --- a/examples/plot_gromov_barycenter.py +++ b/examples/plot_gromov_barycenter.py @@ -17,7 +17,6 @@ computation in POT. import numpy as np
import scipy as sp
-import scipy.ndimage as spi
import matplotlib.pylab as pl
from sklearn import manifold
from sklearn.decomposition import PCA
@@ -90,10 +89,10 @@ def im2mat(I): return I.reshape((I.shape[0] * I.shape[1], I.shape[2]))
-square = spi.imread('../data/square.png').astype(np.float64)[:, :, 2] / 256
-cross = spi.imread('../data/cross.png').astype(np.float64)[:, :, 2] / 256
-triangle = spi.imread('../data/triangle.png').astype(np.float64)[:, :, 2] / 256
-star = spi.imread('../data/star.png').astype(np.float64)[:, :, 2] / 256
+square = pl.imread('../data/square.png').astype(np.float64)[:, :, 2] / 256
+cross = pl.imread('../data/cross.png').astype(np.float64)[:, :, 2] / 256
+triangle = pl.imread('../data/triangle.png').astype(np.float64)[:, :, 2] / 256
+star = pl.imread('../data/star.png').astype(np.float64)[:, :, 2] / 256
shapes = [square, cross, triangle, star]
|