diff options
| author | D.J. Sutherland <djs@djsutherland.ml> | 2020-05-13 15:16:21 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-05-13 22:16:21 +0200 |
| commit | 1511ecc67692773f016cc94190785faa33eab2d5 (patch) | |
| tree | 1a394bd357cc71e2c139507a49de213b414ce075 /examples/gromov/plot_gromov_barycenter.py | |
| parent | 02adb61543bd92158bc2a3193e3c98fd026a6804 (diff) | |
rename I to img in examples, since flake8 complains now (#176)
Diffstat (limited to 'examples/gromov/plot_gromov_barycenter.py')
| -rwxr-xr-x | examples/gromov/plot_gromov_barycenter.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/gromov/plot_gromov_barycenter.py b/examples/gromov/plot_gromov_barycenter.py index f6f031a..e2d88ba 100755 --- a/examples/gromov/plot_gromov_barycenter.py +++ b/examples/gromov/plot_gromov_barycenter.py @@ -84,9 +84,9 @@ def smacof_mds(C, dim, max_iter=3000, eps=1e-9): # The four distributions are constructed from 4 simple images
-def im2mat(I):
+def im2mat(img):
"""Converts and image to matrix (one pixel per line)"""
- return I.reshape((I.shape[0] * I.shape[1], I.shape[2]))
+ return img.reshape((img.shape[0] * img.shape[1], img.shape[2]))
square = pl.imread('../../data/square.png').astype(np.float64)[:, :, 2]
|