Skip to content

Commit

Permalink
Fix speckles in dataset viz
Browse files Browse the repository at this point in the history
I think matplotlib somehow got upgraded and it made the
training chips get speckly and I applied this fix:
matplotlib/matplotlib#17732
  • Loading branch information
lewfish committed Jul 3, 2020
1 parent c5556de commit 235fa77
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -141,4 +141,10 @@ def plot_xyz(self, ax, x, y, z=None):
colors = [tuple([_c / 255 for _c in c]) for c in colors]
cmap = matplotlib.colors.ListedColormap(colors)
labels = labels.numpy()
ax.imshow(labels, alpha=0.4, vmin=0, vmax=len(colors), cmap=cmap)
ax.imshow(
labels,
alpha=0.4,
vmin=0,
vmax=len(colors),
cmap=cmap,
interpolation='none')

0 comments on commit 235fa77

Please sign in to comment.