save-img-in-plt

1
    img_plot = plt.imshow(color) # color is np.array
2
    plt.savefig('foo.png')
3
    plt.show()