国庆头像不用求人，自己生成

国庆节了，用OpenCV实现渐变版国庆头像，用技术给给自己换个国庆版头像！
生成的效果，头像比例有点不好，导致有点压缩。


实现这个其实超级简单，只需要两步！
第一步，找一张国旗模板图象，然后生成mask模板;
第二步，选择一张头像图象，根据mask权重与模板图象重新像素叠加混合。就生成了国庆版渐变头像了;

Python代码（需要安装opencv-python）：

1. import cv2 as cv
   
2. import numpy as np
   
3. 
   
4. def auto_mask_demo(image):
   
5.     gq = cv.imread("C:/Users/Criss/Desktop/gq.png")
   
6.     cv.imshow("input", gq)
   
7. 
   
8.     # 生成mask区域
   
9.     h, w, c = gq.shape
   
10.     mask = np.zeros((h, w), dtype=np.float32)
    
11.     for row in range(h):
    
12.             start = int(np.abs(np.sin(row * np.pi / 180) * 50))
    
13.             mask[row, start:w] = np.linspace(0, 255, w-start)
    
14.     mask = mask / 255.0
    
15.     cv.imshow("mask", mask)
    
16. 
    
17.     # generate icon
    
18.     dst = cv.resize(image, (w, h), interpolation=cv.INTER_CUBIC)
    
19.     result = np.zeros_like(dst)
    
20.     for row in range(h):
    
21.         for col in range(w):
    
22.             w2 = mask[row, col]
    
23.             w1 = 1.0 - w2
    
24.             b1, g1, r1 = gq[row, col]
    
25.             b2, g2, r2 = dst[row, col]
    
26.             b = b1 * w1 + b2 * w2
    
27.             g = g1 * w1 + g2 * w2
    
28.             r = r1 * w1 + r2 * w2
    
29.             result[row, col] = (int(b), int(g), int(r))
    
30.     cv.imshow("profile image", result)
    
31.     cv.imwrite("D:/result.png", result)
    
32.     cv.waitKey(0)
    
33.     cv.destroyAllWindows()
    
34. 
    
35. if __name__ == "__main__":
    
36.     src = cv.imread("C:/Users/Criss/Desktop/headimage.png")
    
37.     auto_mask_demo(src)

复制代码

代码中图像下载：
Img.zip (2 MB, 下载次数: 9)

2021-9-30 17:03 上传

点击文件名下载附件

原文来自【OpenCV学堂】,代码测试后稍作了修改。