Replace cubehelix with the viridis colormap in the RAG drawing example

This commit is contained in:
John Wiggins
2015-08-28 22:17:21 +02:00
parent ddf122f34b
commit 2a6e7ffdfc
+2 -2
View File
@@ -31,8 +31,8 @@ plt.title("RAG with edge weights less than 30, color "
plt.imshow(out)
plt.figure()
plt.title("All edges drawn with cubehelix colormap")
cmap = plt.get_cmap('cubehelix')
plt.title("All edges drawn with viridis colormap")
cmap = plt.get_cmap('viridis')
out = graph.draw_rag(labels, g, img, colormap=cmap,
desaturate=True)