1. 程式人生 > >OpenCV通過隨機數生成圖片

OpenCV通過隨機數生成圖片

import cv2
import numpy
import os

randomByteArray = bytearray(os.urandom(300*400))
flatNumpyArray = numpy.array(randomByteArray)

grayImage = flatNumpyArray.reshape(300, 400)
cv2.imwrite('./out/RandomGray.png', grayImage)

bgrImage = flatNumpyArray.reshape(100, 400, 3)
cv2.imwrite('./out/RandomColor.png', bgrImage)
  • os.urandom是通過系統函式獲取的隨機數
  • reshape(300, 400) 產生灰度圖
  • flatNumpyArray.reshape(100, 400, 3)比上面多了一個引數,3,這個就是RGB資料。