Skip to content
Snippets Groups Projects

Update VAE and fidle

Former-commit-id: 8546f434
parent 70fb7cc2
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
...@@ -155,11 +155,11 @@ def rmin(l): ...@@ -155,11 +155,11 @@ def rmin(l):
# show_images # show_images
# ------------------------------------------------------------- # -------------------------------------------------------------
# #
def plot_images(x,y=None, indices=None, columns=12, x_size=1, y_size=1, colorbar=False, y_pred=None, cm='binary'): def plot_images(x,y=None, indices='all', columns=12, x_size=1, y_size=1, colorbar=False, y_pred=None, cm='binary'):
""" """
Show some images in a grid, with legends Show some images in a grid, with legends
args: args:
X: images - Shapes must be (-1 lx,ly,1) or (-1 lx,ly,3) x: images - Shapes must be (-1,lx,ly) (-1,lx,ly,1) or (-1,lx,ly,3)
y: real classes or labels or None (None) y: real classes or labels or None (None)
indices: indices of images to show or None for all (None) indices: indices of images to show or None for all (None)
columns: number of columns (12) columns: number of columns (12)
...@@ -177,8 +177,7 @@ def plot_images(x,y=None, indices=None, columns=12, x_size=1, y_size=1, colorbar ...@@ -177,8 +177,7 @@ def plot_images(x,y=None, indices=None, columns=12, x_size=1, y_size=1, colorbar
draw_labels = (y is not None) draw_labels = (y is not None)
draw_pred = (y_pred is not None) draw_pred = (y_pred is not None)
# if y_pred is None: y_pred=y if indices=='all': indices=range(len(x))
if indices==None: indices=range(len(x))
for i in indices: for i in indices:
axs=fig.add_subplot(rows, columns, n) axs=fig.add_subplot(rows, columns, n)
n+=1 n+=1
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment