Numpy 2 Image

import numpy as np
import cv2
r = np.reshape(np.arange(256*256)%256,(256,256)) 

g = np.zeros_like(r)

b = r.T

cv2.imwrite('/Users/rajacsp/datasets/barcode_images/gradients.png', np.dstack([b,g,r])) 

print('Done')
Done