Save img in plt

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