前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >opencv+python制作硬核七夕礼物

opencv+python制作硬核七夕礼物

作者头像
AI算法与图像处理
发布2020-08-28 16:25:14
9000
发布2020-08-28 16:25:14
举报

重磅干货,第一时间送达

来源:OpenCV视觉实践

明天就是七夕了!

如果有对象,祝早生贵子!没有对象,祝早结连理!

(搞笑一下,祝大家幸福美满吧,嘿嘿嘿)

给大家准备了一份已经写好的代码,利用python+opencv+openpyxl,opencv实现读取图片像素颜色,然后openpyxl自动填充Excel表格对应单元格背景颜色,得到一副Excel填充图画。文末代码下载链接。

先来看看效果叭(gif动图)!

把图片换成女朋友的真人图像会更逼真,会更加更好看。

然后看代码叭!

配置环境

我们需要用到python以及openpyxl和opencv包,没有下载的小伙伴可以去配置一下,超级简单:

大概就是在python终端(看个人环境)执行:

代码语言:javascript
复制
pip install opencv-python
pip install openpyxl

下面是我安装openpyxl包的录屏,可以参考一下,也可以跳过。

配置完成后就是导入我们需要的包了,这儿不需要过多了解:

代码语言:javascript
复制
#导入opencv和openpyxl
import openpyxl
import cv2 as cv
import numpy as np
from openpyxl import load_workbook
from openpyxl import workbook
# 导入字体、边框、颜色以及对齐方式相关库
from openpyxl.styles import Font, Border, Side, PatternFill, colors, Alignment

读取Excel表格

我们首先需要读取一个xlsx格式的Excel表格,需要提前在工作路径或者某绝对路径进行创建,并自定义命名,然后我们得到了一个表格对象sheet,就可以对它进行读写设置等操作了。

代码语言:javascript
复制
#设置路径
#file_path = r"L:\jupyter notebook_code\file\zhaopian.xlsx"  #绝对路径
# 加载工作簿
wb = load_workbook(filename='zhaopian.xlsx') #默认工作路径,zhaopian.xlsx为Excel表格名称,自行更改
# 获取活动中的表格
sheet = wb.active
print("open sheet:", sheet.title) #打印表格标题
print("row:", sheet.max_row, "column:", sheet.max_column)

opencv读取图像并显示

代码语言:javascript
复制
#opencv读取图像并显示
'''
opencv的imread函数读取图片到srcImage
如果按原始大小读取图片,可以改为:srcImage = cv.imread('zhu.jpg')
加上第二个参数33:srcImage = cv.imread('zhu.jpg',33)
表示将原图缩为原来的1/4进行读取。当原图过大是可以采用
'''
srcImage = cv.imread('zhaopian.jpg') #zhu.jpg为照片名字,自行更改
cv.imshow('image',srcImage)

#获取图像的高,宽并打印
height = srcImage.shape[0]
width = srcImage.shape[1]
print("height:", height, "width:", width)

利用opencv将图像读取进来,之后我们就要利用for循环遍历图像的各个像素值了,但在这之前我们需要自定义一个函数,因为我们是用openpyxl包的

代码语言:javascript
复制
PatternFill(fill_type='solid',fgColor= 'FFFFFF')

函数填充单元格颜色的,该函数只能设置十六进制的颜色格式:

所以我们需要自己写一个函数将RGB颜色格式转换为十六进制:

代码语言:javascript
复制
#自定义函数,将RGB颜色值转换为16进制的字符串格式
def color_transform(value):
    digit = list(map(str, range(10))) + list("ABCDEF")
    if isinstance(value, tuple):
        string = ''
        for i in value:
            a1 = i // 16
            a2 = i % 16
            string += digit[a1] + digit[a2]
        return string
    elif isinstance(value, str):
        a1 = digit.index(value[1]) * 16 + digit.index(value[2])
        a2 = digit.index(value[3]) * 16 + digit.index(value[4])
        a3 = digit.index(value[5]) * 16 + digit.index(value[6])
        return (a1, a2, a3)

该函数实现的功能

代码语言:javascript
复制
输入:(0,0,0),输出:000000
输入:#FFFFFF,输出:(255, 255, 255)

好了,解决了颜色转换,我们就遍历像素吧!

代码语言:javascript
复制
#遍历循环图片,获取各像素点位置的像素值,并利用openpyxl改写对应位置单元格颜色
for y in range(1, height):
    print("第几行:",y) #供查看进度
    for x in range(1, width):
        Blue_value = srcImage[y,x,0] #获取蓝,绿,红三个通道
        Green_value = srcImage[y,x,1]
        Red_value = srcImage[y,x,2]
        color = (Red_value,Green_value,Blue_value) #转为元组并附给color_transform函数得到16进制
        hex_ = color_transform(color)
        fille = PatternFill(fill_type='solid',fgColor= hex_) #设置颜色
        sheet.cell(y,x).fill=fille #openpyxl进行颜色填充

具体大家看注释了,大概就是遍历像素并读取RGB像素值,然后传给color_transform函数进行转换,然后赋值给openpyxl的PatternFill进行Excel表格填充。

还有一步,就是保存了:

代码语言:javascript
复制
wb.save(filename = 'zhaopian.xlsx')  # 保存工作簿
print("保存成功")

cv.waitKey(30)#图片展示30ms
cv.destroyAllWindows()

完整代码下载链接:

链接: https://pan.baidu.com/s/1QOw28b-8tQpUp2j0qz26rg 提取码:02q9

最后一步!超重要

上面程序完毕之后,我们只是完成了单元格的填充,但我们知道,单元格不是正方形,运行完图片是扁平的:

所以还需要我们手动更改一下单元格的大小,其实openpyxl也是可以做到的,但我也是第一天用openpyxl,也不熟练,能动手就动手吧,毕竟最麻烦的已经完成了。

还有还有!在发给对象之前,一定把excel表格打开看一下,确保是这样的:

也就是缩放正常的表格,只有这样,当她滚动鼠标滚轮进行缩放,渐渐变成下图这样时,才会有惊喜嘛!

如果直接一打开就是上图,和一个照片也没什么两样,也就没那么惊喜了,是不是?

还有,别告诉你对象是用代码生成的,告诉她:

“这是我一个一个单元格填充了一个月颜色的成果,我是爱你的,么么哒,带上身份证,咱们去看电影吧~”

THE END

兄弟们,看在我熬夜调BUG的份上!转发到你们空间或者群聊吧!

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-08-25,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 AI算法与图像处理 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档