AI学习者 · 2021年10月08日

OpenCV用20行代码生成国庆渐变版头像

国庆节了,用OpenCV实现渐变版国庆头像,用技术给给自己换个国庆版头像!

实现过程:
实现这个其实超级简单,只需要两步!

第一步,找一张国旗模板图象,然后生成mask模板,图示如下:
image.png

第二步,选择一张头像图象,根据mask权重与模板图象重新像素叠加混合。就生成了国庆版渐变头像了。
image.png

源码如下:

import cv2 as cv
import numpy as np


def auto_mask_demo(image):
    gq = cv.imread("D:/2.png")
    cv.imshow("input", gq)

    # 生成mask区域
    h, w, c = gq.shape
    mask = np.zeros((h, w), dtype=np.float32)
    for row in range(h):
            start = np.int(np.abs(np.sin(row * np.pi / 180) * 50))
            mask[row, start:w] = np.linspace(0, 255, w-start)
    mask = mask / 255.0
    cv.imshow("mask", mask)

    # generate icon
    dst = cv.resize(image, (w, h), interpolation=cv.INTER_CUBIC)
    result = np.zeros_like(dst)
    for row in range(h):
        for col in range(w):
            w2 = mask[row, col]
            w1 = 1.0 - w2
            b1, g1, r1 = gq[row, col]
            b2, g2, r2 = dst[row, col]
            b = b1 * w1 + b2 * w2
            g = g1 * w1 + g2 * w2
            r = r1 * w1 + r2 * w2
            result[row, col] = (np.int(b), np.int(g), np.int(r))
    cv.imshow("profile image", result)
    cv.imwrite("D:/result.png", result)
    cv.waitKey(0)
    cv.destroyAllWindows()

if __name__ == "__main__":
    src = cv.imread("D:/1.png")
    auto_mask_demo(src)
原文:OpenCV学堂
作者:gloomyfish

推荐阅读

推荐阅读
关注数
18808
内容数
1351
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:aijishu20(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息