python生成测试图片

悠悠 2021-12-21 00:17 402阅读 0赞

直接代码

1 import cv2.cv as cv
     2 saveImagePath = 'E:/ScreenTestImages/'
     3 
     4 colorRed = [0,0,255]
     5 colorGreen = [0,255,0]
     6 colorBlue = [255,0,0]
     7 colorWhite = [255,255,255]
     8 colorBlack = [0,0,0]
     9 colorAqua = [255,255,0]
    10 colorFuchsia = [255,0,255]
    11 colorYellow = [0,255,255]
    12 
    13 stardardColors = [colorBlue,colorGreen,colorAqua,colorRed,colorFuchsia,colorYellow,colorWhite]
    14 
    15 def createImg(depth=3):
    16     return cv.CreateImage((800,480),8,depth)
    17 
    18 def saveImageFile(typeName,img):
    19     filename = saveImagePath + typeName + '.png'
    20     cv.SaveImage(filename,img)
    21     print typeName+'.png','\t\t...\tok'
    22 
    23 def createOneColorImage(color):
    24     img = createImg()
    25     cv.Set(img,color)
    26     return img
    27 
    28 def create64GrayImage():
    29     img = createImg(1)
    30     cv.SetZero(img)
    31     for xPos in range(0,64):
    32         cv.SetImageROI(img,(int(12.5*xPos),0,800,480))
    33         cv.Set(img,xPos*255/63)
    34         cv.ResetImageROI(img)
    35     return img
    36 
    37 def createCheckBoardPattern(isReserved=False):
    38     img = createImg(1)
    39     boolColor = True
    40     numsX = 4
    41     numsY = 4
    42     pixsX = 800/numsX
    43     pixsY = 480/numsY
    44     for x in range(0,numsX):
    45         for y in range(0,numsY):
    46             cv.SetImageROI(img,(x*pixsX,y*pixsY,(x+1)*pixsX,(y+1)*pixsY))
    47             boolColor = not (x%2) ^ (y%2) ^ isReserved            
    48             cv.Set(img,255*boolColor)
    49             cv.ResetImageROI(img)           
    50     return img
    51 
    52 def createStardardImage():
    53     img = createImg()
    54     pixs = 800./7
    55     for i in range(0,7):
    56         cv.SetImageROI(img,(int(i*pixs),0,int(i*pixs+pixs),480))
    57         cv.Set(img,stardardColors[i])
    58         cv.ResetImageROI(img)
    59     return img
    60 
    61 if __name__ == '__main__':
    62     print 'Start Gen Test Screen Files ...'
    63     saveImageFile('red',createOneColorImage(colorRed))
    64     saveImageFile('green',createOneColorImage(colorGreen))
    65     saveImageFile('blue',createOneColorImage(colorBlue))    
    66     saveImageFile('white',createOneColorImage(colorWhite))
    67     saveImageFile('black',createOneColorImage(colorBlack))
    68     saveImageFile('64gray',create64GrayImage())
    69     saveImageFile('checkboard Pattern',createCheckBoardPattern())
    70     saveImageFile('checkboard Pattern(inverted)',createCheckBoardPattern(True))
    71     saveImageFile('standard',createStardardImage())
    72     print 'Generate Success!'
    73

贴上非单色图片的样子：

64 gray:

![141618392242352.png][]

checkboard pattern:

![141620197249048.png][]

checkboard pattern(inverted):

![141618594287380.png][]

standard:

![141620525684882.png][]

转载于:https://www.cnblogs.com/zhangyonghugo/p/3842951.html

[141618392242352.png]: /images/20211220/2a805fee93314a33ac4a28496853cbca.png
[141620197249048.png]: /images/20211220/36f2f4f5b99745f7b30a948a418a77b3.png
[141618594287380.png]: /images/20211220/8eb666fdce4543a984dffb1e7c8e5b0e.png
[141620525684882.png]: /images/20211220/4774af4629a1411d80ab345b348f0233.png