直接上代码吧：
import cv2.cv as cv
saveImagePath = 'E:/ScreenTestImages/'
colorRed = [0,0,255]
colorGreen = [0,255,0]
colorBlue = [255,0,0]
colorWhite = [255,255,255]
colorBlack = [0,0,0]
colorAqua = [255,255,0]
colorFuchsia = [255,0,255]
colorYellow = [0,255,255]
stardardColors = [colorBlue,colorGreen,colorAqua,colorRed,colorFuchsia,colorYellow,colorWhite]
def createImg(depth=3):
return cv.CreateImage((800,480),8,depth)
def saveImageFile(typeName,img):
filename = saveImagePath + typeName + '.png'
cv.SaveImage(filename,img)
print typeName+'.png','\t\t...\tok'
def createOneColorImage(color):
img = createImg()
cv.Set(img,color)
return img
def create64GrayImage():
img = createImg(1)
cv.SetZero(img)
for xPos in range(0,64):
cv.SetImageROI(img,(int(12.5*xPos),0,800,480))
cv.Set(img,xPos*255/63)
cv.ResetImageROI(img)
return img
def createCheckBoardPattern(isReserved=False):
img = createImg(1)
boolColor = True
numsX = 4
numsY = 4
pixsX = 800/numsX
pixsY = 480/numsY
for x in range(0,numsX):
for y in range(0,numsY):
cv.SetImageROI(img,(x*pixsX,y*pixsY,(x+1)*pixsX,(y+1)*pixsY))
boolColor = not (x%2) ^ (y%2) ^ isReserved
cv.Set(img,255*boolColor)
cv.ResetImageROI(img)
return img
def createStardardImage():
img = createImg()
pixs = 800./7
for i in range(0,7):
cv.SetImageROI(img,(int(i*pixs),0,int(i*pixs+pixs),480))
cv.Set(img,stardardColors[i])
cv.ResetImageROI(img)
return img
if __name__ == '__main__':
print 'Start Gen Test Screen Files ...'
saveImageFile('red',createOneColorImage(colorRed))
saveImageFile('green',createOneColorImage(colorGreen))
saveImageFile('blue',createOneColorImage(colorBlue))
saveImageFile('white',createOneColorImage(colorWhite))
saveImageFile('black',createOneColorImage(colorBlack))
saveImageFile('64gray',create64GrayImage())
saveImageFile('checkboard Pattern',createCheckBoardPattern())
saveImageFile('checkboard Pattern(inverted)',createCheckBoardPattern(True))
saveImageFile('standard',createStardardImage())
print 'Generate Success!'效果：
