画一个奥利奥(python+opencv)

来自定义属于自己的奥利奥吧!

惯例先放效果图:
画一个奥利奥(python+opencv)
原项目作者:Diu
灵感和图片来源:https://github.com/ddiu8081/oreooo
本项目仅为原作者项目的python复刻版本(原作者用的是vue)
饼和馅使用的是原作者提供的图片:https://github.com/ddiu8081/oreooo/tree/master/assets/image
请在阅读原作者的 教程 并理解思路以后,再阅读以下代码,不然可能看不懂
本文中用到的opencv相关函数不作详细解释,有兴趣可自行搜教程学习
(解释起来太罗嗦了_(:з)∠)_)
环境要求:下图中的东西要在同个文件夹,其它的没啥要求
画一个奥利奥(python+opencv)
图片下载链接,提取码: t2zs
话不多说放代码(注释应该够详细了,还有问题可在评论区提出):

import cv2

def init(): #  将加载本地图片作为函数封装起来,以便后续作为模块使用
    imge1 = cv2.cv2.imread('O.png', cv2.IMREAD_UNCHANGED)  # 上半饼
    imge2_temp = cv2.imread('R.png', cv2.IMREAD_UNCHANGED)  # 暂时的馅,后续要做缩小处理
    imge3 = cv2.imread('Ob.png', cv2.IMREAD_UNCHANGED)  # 下半饼
    
    # 空白画布,在最底层为馅的时候要用
    imge_empty = cv2.imread('empty.png', cv2.IMREAD_UNCHANGED)  

    # 对馅进行处理,缩小到90%,毕竟总不能馅和饼一样大
    scale_percent = 90
    width = int(imge2_temp.shape[1] * scale_percent / 100)
    height = int(imge2_temp.shape[0] * scale_percent / 100)
    imge2 = cv2.resize(imge2_temp, (width, height), interpolation=cv2.INTER_AREA)
    return imge1, imge2, imge3, imge_empty  # 将上半饼,缩小的馅,下半饼以及空白画布作为对象返回


#  画布增加(为了让图片能叠加,和ps一个道理)
def png_extend(img, px):
    # 增加有颜色的像素,value的3个值代表RGB,随便啥都行(反正后续要变成透明),这里为白色
    imgb = cv2.copyMakeBorder(img, px, 0, 0, 0, cv2.BORDER_CONSTANT, value=[255, 255, 255])
    # 分离4个通道(R,G,B和Alpha(透明度))(虽然平时念RGB比较习惯,不过在opencv里面顺序变成BGR了)
    b_channel, g_channel, r_channel, alpha_channel = cv2.split(imgb)  
    alpha_channel[:px, :] = 0  # 把有颜色的像素变透明
    return imgb  # 处理完的画布作为整体返回


def add_t(imgb):  # 增加上半饼,只在要叠加最上面一层的时候使用
    img1, img2, img3, img_empty = init()  # 初始化本地图片
    roi = imgb[0:410, 0:600]  # 设置要叠加的区域

	# 下面的步骤为opencv中教科书般的“按位运算”操作,和百度能找到的教程几乎一样
    img1gray = cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY)
    
    # 这里的第二个参数253不是唯一可用的值,可在255以内随意尝试
    # 不过太低会导致图像识别差异过大,我试过的最低大概是248左右
    ret, mask = cv2.threshold(img1gray, 253, 255, cv2.THRESH_BINARY)  
    
    mask_inv = cv2.bitwise_not(mask)
    img4_bg = cv2.bitwise_and(roi, roi, mask=mask)
    img1_fg = cv2.bitwise_and(img1, img1, mask=mask_inv)
    dst = cv2.add(img4_bg, img1_fg)
    imgb[0:410, 0:600] = dst
    return imgb  # 处理完的画布作为整体返回


def add_re(imgb):  # 注释同函数add_t
    img1, img2, img3, img_empty = init()
    roi = imgb[0:369, 30:570]
    regray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
    ret, mask = cv2.threshold(regray, 253, 255, cv2.THRESH_BINARY)
    mask_inv = cv2.bitwise_not(mask)
    imgb_bg = cv2.bitwise_and(roi, roi, mask=mask)
    re_fg = cv2.bitwise_and(img2, img2, mask=mask_inv)
    dst = cv2.add(imgb_bg, re_fg)
    imgb[0:369, 30:570] = dst
    return imgb


def add_b(imgb):  # 注释同函数add_t
    img1, img2, img3, img_empty = init()
    roi = imgb[0:410, 0:600]
    img1gray = cv2.cvtColor(img3, cv2.COLOR_BGR2GRAY)
    ret, mask = cv2.threshold(img1gray, 253, 255, cv2.THRESH_BINARY)
    mask_inv = cv2.bitwise_not(mask)
    img4_bg = cv2.bitwise_and(roi, roi, mask=mask)
    img1_fg = cv2.bitwise_and(img3, img3, mask=mask_inv)
    dst = cv2.add(img4_bg, img1_fg)
    imgb[0:410, 0:600] = dst
    return imgb


def draw(name):  # 这里传进来的的name是你定义的奥利奥字段,如“奥利奥奥利利利奥”
	# 预处理
    img1, img2, img3, img_empty = init()
    if name[-1] == '奥':  # 如果最底层是“奥”,则直接用下半饼作为基础画布
        cv2.imwrite('img4.png', img3)
    else:  # 如果最底层是“利”,则要在空白画布上附加缩小后的馅,再作为基础画布,以便后续图片宽度不变
        img4 = add_re(img_empty)
        cv2.imwrite('img4.png', img4)    
    img4 = cv2.imread('img4.png', cv2.IMREAD_UNCHANGED)  # img4就是我们的基础画布了

	# 对除去顶层以外的部分进行叠图(因为顶层有可能要叠上半饼,所以后续拉出来单独处理)
    for i in range(0, len(name) - 2):
        if (name[len(name) - i - 1] == '奥') & (name[len(name) - i - 2] == '利'):
            """底+馅要拓展40像素"""
            imgt = png_extend(img4, 40)
            img4 = add_re(imgt)
        elif (name[len(name) - i - 1] == '利') & (name[len(name) - i - 2] == '利'):
            """馅+馅要拓展60像素"""
            img4 = png_extend(img4, 60)
            img4 = add_re(img4)
        elif (name[len(name) - i - 1] == '利') & (name[len(name) - i - 2] == '奥'):
            """馅+底/顶要拓展84像素"""
            img4 = png_extend(img4, 84)
            img4 = add_b(img4)
        elif (name[len(name) - i - 1] == '奥') & (name[len(name) - i - 2] == '奥'):
            """底+底/顶要拓展64像素"""
            img4 = png_extend(img4, 64)
            img4 = add_b(img4)
    
    # 对顶层单独处理
    if (name[0] == '奥') & (name[1] == '利'):
        img4 = png_extend(img4, 84)
        img4 = add_t(img4)
    elif (name[0] == '奥') & (name[1] == '奥'):
        img4 = png_extend(img4, 64)
        img4 = add_t(img4)
    elif (name[0] == '利') & (name[1] == '奥'):
        imgt = png_extend(img4, 40)
        img4 = add_re(imgt)
    elif (name[0] == '利') & (name[1] == '利'):
        img4 = png_extend(img4, 60)
        img4 = add_re(img4)
    cv2.imwrite('oreo.png', img4)  # 将最终图像保存为oreo.png
    # 以下被注释的代码用来临时调试使用,可将最终图片自动展示出来,不用手动点开图片查看
    # cv2.imshow('image', img4)
    # cv2.waitKey(0)  
    # cv2.destroyAllWindows()

被其它py文件调用只需import oreo,然后oreo.draw('奥利利利奥')即可

上一篇:与matlab里面 imreconstruct 函数相同的python代码


下一篇:OpenCV图像拼接-Stitcher类-Stitching detailed使用与参数介绍OpenCV常用图像拼接方法(一):直接拼接(硬拼)