Loading [MathJax]/jax/output/CommonHTML/config.js
前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >如何用 Python 给照片换色

如何用 Python 给照片换色

作者头像
机器视觉CV
发布于 2019-10-20 10:53:51
发布于 2019-10-20 10:53:51
3.4K03
代码可运行
举报
文章被收录于专栏:机器视觉CV机器视觉CV
运行总次数:3
代码可运行

最近遇到了一个需求,就是对图片进行色彩风格转换,让一个物体可以以各种不同的色彩来呈现。

比如一个红色的苹果,我想把它转化成绿色,这可怎么办呢?本来想的解决方案是先识别边界,然后对边界内区域进行色彩替换或者填充,这样整个流程就分成了两步,首先需要进行边界判断,有了边界之后才能对某些区域颜色进行替换填充,填充的区域还是不规则的,填充的颜色也需要根据实际的情况来变化,同时还要兼顾阴影、高光等等的处理,想想真是头大。

然后我就突然想到了之前学 PS 的时候,曾经做过对某一张图片进行色调替换,我还记得当时调的内容是「色相」,通过调节这个值可以实现各种色调的转换。这之后我就心想,能不能用程序来做这件事呢?

于是我就开始了对图像色彩的一些研究,研究完了之后就实现了用 Python 来更换图片色调的功能,将这篇文章顺便记录下来。

先给大家看看效果吧,就比如我从百度上随便搜一张图,比如热气球吧,是这样子:

原图

我把它转成红色、黄色、蓝色、紫色,基本就是这个样子:

红色风格

黄色风格

绿色图片

蓝色图片

这里的图片都是经过 Python 自动转换算法实现的,主要调节的就是色相。其实代码实现是比较简单的,但在实现之前需要了解一些图像色彩知识。了解了这些之后我们再实现才会更加游刃有余。

下面我们就首先来了解一下图像色彩的基本知识,然后用 Python 实现色调转换算法吧。

RGB

首先让我们来了解下颜色的三原色,RGB。

RGB,其实就是三种颜色,分别代表红色(Red)、绿色(Green)、蓝色(Blue),用这三种原色颜色混合可以表示任意的颜色。RGB 是根据颜色发光的原理来设计的,比如这里有红绿蓝三道光,当三束光混合在一起的时候,其呈现的最终的光效颜色就取决于这三种原色光的强弱了。

•比如说红光最强,绿光和蓝光几乎没有,那么最后结果肯定呈现为红光了。•如果蓝光很弱,红光和绿光非常强,那么结果就是红光和绿光的混合光,也就黄光。•如果红绿蓝三种光都非常强,那么就会呈现三种光的混合光,也就是白光。•如果三种光都非常弱,那就几乎没有光,自然就是黑色了。

如果大家了解过 RGB 的一些知识的话,想必也见过这张混合光图:

但是这仅仅是一张平面图,并不能很好地体现三原色的混合关系,大家可以看这张图:

实际上可以对 RGB 色值建立一个三维坐标系,坐标原点就在上图中所看不见的那个正方体顶点,向左延伸为 R 红色轴,向右延伸为 B 蓝色轴,向上延伸为 G 绿色轴。

RGB 在进行色彩表示时使用了 256 阶,也就是从 0-255,可以由一个字节来表示。数值越大,RGB 三个轴,每个轴对应的数值越大,就代表其亮度越高,最亮就对应着 255,最暗就对应着 0。三个轴上的三个数字联合起来可以用来表示一个颜色。

例如:

•(255, 255, 255) 就代表 RGB 都是满的,组成白色。•(255, 255, 0) 就代表 R 红色是满的,G 绿色是满的,B 蓝色是没有的,红色和绿色混合为黄色,所以它就表示黄色。•(0, 0, 0) 就代表 RGB 都是没有的,呈现黑色。

因此,通过这三个数值,我们就可以实现任何颜色的表示了。

HSV

但了解了 RGB 显然不能应对我们的需求,它和色相也没有什么关系,所以这里我们还需要研究另外一个颜色模型,叫做 HSV。

HSV,又叫做 HSB,其实也是三个参数,分别是色调(Hue)、饱和度(Saturation)、明度(Brightness),在 HSV 中 V 代表 Value,也是明度的意思。通过这三个值,我们同样可以表示任意的颜色。

首先我们看看 HSV (HSB) 颜色模型的坐标轴图吧,它可以用这么一个锥形的坐标来表示:

色调,Hue,它是一种角度度量,就是图中 Hue 所示的角度,绕圆锥体一周。它的值是从红色开始逆时针方向计算的,红色是 0,绿色是 120,蓝色为 240,形成一个红绿蓝三原色组成的色带。

饱和度,Saturation,它是图中 Saturation 所示的方向,由最上面圆的圆心向外扩散。它表示颜色接近光谱色的程度。一种颜色,可以看成是某种光谱色与白色混合的结果。其中光谱色所占的比例愈大,颜色接近光谱色的程度就愈高,颜色的饱和度也就愈高。饱和度高,颜色则深而艳。光谱色的白光成分为 0,饱和度达到最高。通常取值范围为 0-100,值越大,颜色越饱和。也就是说图中,圆心处饱和度为 0,越靠近边,饱和度越高,最高达到 100。

明度,Brightness,它表示颜色明亮的程度,对于光源色,明度值与发光体的光亮度有关,对于物体色,此值和物体的透射比或反射比有关。通常取值范围为 0-100,0 对应黑色,100 对应白色。

另外大家如果用到取色板的话,它是这样子的:

你可以看到左边有个大的矩形,右边有个竖直的彩虹带。由于刚才所说的圆锥坐标不好表示,在调色板里面就把圆锥坐标拆成了两部分。

•右边的彩虹带就可以选择色调 Hue,拖动彩虹带的箭头位置就可以调整 Hue 的值。•左边的正方形就和饱和度 Saturation、明度 Brightness 有关,横坐标表示饱和度 Saturation,纵坐标表示 Brightness,从左到右,饱和度 Saturation 变高,从下到上,明度 Brightness 变高。•左下角,由于饱和度 Saturation 和明度 Brightness 都为 0,就呈现黑色。•左上角,由于饱和度 Saturation 为 0,所以相当于没有颜色,颜色最浅,但是它的明度 Brightness 达到了最高,所以它就显示没有颜色的明度最高的色,即白色。•右上角,由于饱和度 Saturation 和明度 Brightness 都为最高,那就显示最纯的 Hue 值。

这也就是 HSB 颜色模型的原理。

同样地,我们可以用 HSB 的三个值来表示任意的颜色,因此 HSB 也成为表示颜色的基本标准之一。

HSB 和 RGB 的转换

HSB 和 RGB 都能表示一个颜色,它们之间也是可以相互转换的,可以一一对应。

他们之间的转换逻辑这里就不再专门对其公式展开详解了,公式总结如下:

RGB to HSB

HSV to RGB

大家如果感兴趣想要研究的话也可以去:https://www.rapidtables.com/convert/color/hsv-to-rgb.html 来了解一下详情,这里还有详细的转换表格以及实时转换实现。

另外对于它们之间的转换算法,很多库都已经封装好了。我们可以直接调用,比如 Python 中的 colorsys 模块,就实现了 rgb_to_hsv 和 hsv_to_rgb 算法,我们也可以直接使用。

色调转换

了解了以上内容之后,我们就可以使用程序来实现色调转换了。相比我们已经知道应该改什么内容了,那就是修改 HSV 中的 H 值,通过不同的 H 值我们就可以将图片转换为不同的色调了。

具体思路是怎样的呢:

•首先获取图像每个像素的的 RGB 色值。•将 RGB 色值转化为 HSV 色值。•调整 HSV 色值中的 H。•将 HSV 色值转回 RGB 色值。•输出图像。

一共就是这么五步,这里的两个转换色值的操作我们就可以借助于 colorsys 模块。另外需要安装一个 pillow 模块:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
pip3 install pillow

安装完毕之后,我们准备任意一张图片,然后实现上面的五个步骤。

最终的代码实现如下:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
import colorsysfrom PIL import Image
# 输入文件filename = 'input.jpg'# 目标色值target_hue = 0
# 读入图片,转化为 RGB 色值image = Image.open(filename).convert('RGB')
# 将 RGB 色值分离image.load()r, g, b = image.split()result_r, result_g, result_b = [], [], []# 依次对每个像素点进行处理for pixel_r, pixel_g, pixel_b in zip(r.getdata(), g.getdata(), b.getdata()):    # 转为 HSV 色值    h, s, v = colorsys.rgb_to_hsv(pixel_r / 255., pixel_b / 255., pixel_g / 255.)    # 转回 RGB 色系    rgb = colorsys.hsv_to_rgb(target_hue, s, v)    pixel_r, pixel_g, pixel_b = [int(x * 255.) for x in rgb]    # 每个像素点结果保存    result_r.append(pixel_r)    result_g.append(pixel_g)    result_b.append(pixel_b)
r.putdata(result_r)g.putdata(result_g)b.putdata(result_b)
# 合并图片image = Image.merge('RGB', (r, g, b))# 输出图片image.save('output.png')

具体的实现在代码注释里面已经很清楚了。

在这里我们将 target_hue 定义为 0。通过前文我们知道,Hue 为 0 代表红色,120 代表绿色,240 代表蓝色。我们可以自定义 0-355 这 360 个数值,实现不同的色调转换。

如果想输出其他颜色的图片,就把 target_hue 这个值改一下就好了。

看看最后的运行效果,输入是一张原图:

所以最后的输出效果就是如下的结果:

如果将代码中的 target_hue 值进行更改,就会呈现不同的颜色风格了,就像文中开头所示的一样。

比如把飞猪的 Logo 由黄色变为红色,都是可以做到的:

处理透明像素

上面的算法仅仅考虑了 RGB,如果有些图包含了透明像素,上面的程序对于透明像素是无法处理的,最后输出的结果会带有某种颜色的背景。

对于透明像素的处理,我们可以增加一个维度的值,就是 A,即 Alpha 透明度。

所以使用 RGBA 和 HSV 的转换我们就可以实现透明像素的处理了,代码实现如下:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
import colorsysfrom PIL import Image
# 输入文件filename = 'input.png'# 目标色值target_hue = 0
# 读入图片,转化为 RGB 色值image = Image.open(filename).convert('RGBA')
# 将 RGB 色值分离image.load()r, g, b, a = image.split()result_r, result_g, result_b, result_a = [], [], [], []# 依次对每个像素点进行处理for pixel_r, pixel_g, pixel_b, pixel_a in zip(r.getdata(), g.getdata(), b.getdata(), a.getdata()):    # 转为 HSV 色值    h, s, v = colorsys.rgb_to_hsv(pixel_r / 255., pixel_b / 255., pixel_g / 255.)    # 转回 RGB 色系    rgb = colorsys.hsv_to_rgb(target_hue, s, v)    pixel_r, pixel_g, pixel_b = [int(x * 255.) for x in rgb]    # 每个像素点结果保存    result_r.append(pixel_r)    result_g.append(pixel_g)    result_b.append(pixel_b)    result_a.append(pixel_a)
r.putdata(result_r)g.putdata(result_g)b.putdata(result_b)a.putdata(result_a)
# 合并图片image = Image.merge('RGBA', (r, g, b, a))# 输出图片image.save('output.png')

在这里就是增加了对透明像素的处理,在图像 convert 和 merge 的时候都使用 RGBA 模式,就可以保留原有图片中的透明像素了。

以上便是使用 Python 程序自动调整色调的实现。

其他应用

另外对于图像调色的应用还有很多,比如我们可以不直接指定 Hue 的值,而是将 Hue 的值在原来的基础上增加或减少,则可以实现彩虹变色,如:

原图是这样子:

经过处理之后我们可以得到这样的效果:

这样我们使用程序处理出各种风格的好看的照片了,是不是很有意思呢?具体的实现可以参考:https://stackoverflow.com/questions/27041559/rgb-to-hsv-python-change-hue-continuously。

大家如果觉得有意思,就快来试试吧~

参考链接:

•https://blog.csdn.net/weixin_43280713/article/details/84133915•https://stackoverflow.com/questions/27041559/rgb-to-hsv-python-change-hue-continuously•https://blog.csdn.net/qq_33759946/article/details/90295377

图片来源于互联网公开内容,如有图片侵权,请联系我删除,谢谢。


个人站点:

Github: https://github.com/FLyingLSJ

CSDN : https://blog.csdn.net/LSJ944830401

BLOG: https://flyinglsj.github.io/

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

本文分享自 机器视觉CV 微信公众号,前往查看

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

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

评论
登录后参与评论
暂无评论
推荐阅读
编辑精选文章
换一批
花了大半个月,我终于逆向分析了Github Copilot
众所周知,Github Copilot是一种基于机器学习的代码自动补全工具。它使用了来自GitHub的大量代码作为训练数据,并使用OpenAI的语言模型来生成代码。Copilot还可以学习用户的编码习惯,并根据上下文推断出正确的代码片段。
孟健
2023/08/17
1.5K0
花了大半个月,我终于逆向分析了Github Copilot
完结撒花,copilot——细节决定成败!
今年五月的时候我写了一篇文章《**花了大半个月,我终于逆向分析了Github Copilot**》,受到了不少关注。当时应该还是copilot刚出不久,通过AST将webpack_modules的chunk自动拆分到不同的module文件,并通过一些语法转换让代码更易读。
孟健
2023/11/16
7010
完结撒花,copilot——细节决定成败!
对Copilot进行逆向工程之后,我发现它可能只用了参数量12B的小模型
2021 年,微软、OpenAI、Github 三家联合打造了一个好用的代码补全与建议工具 ——Copilot。
机器之心
2023/04/06
1.7K0
对Copilot进行逆向工程之后,我发现它可能只用了参数量12B的小模型
再次揭秘Copilot:sourcemap逆向分析
今年五月的时候我写了一篇文章《**花了大半个月,我终于逆向分析了Github Copilot「》,最近发现copilot将.map文件也提交了上来,」sourcemap**中包含了整体源码的结构信息和部分变量信息,这无疑为分析copilot源码带来了极大的便利,因此再次分析一波。
孟健
2023/11/11
3120
再次揭秘Copilot:sourcemap逆向分析
jscodeshift 那点秘密
咱们从入口开始分析,先看看这个 transform 文件的几个参数是什么?这部分代码位于 src/Worker.js:
码农小余
2022/06/16
1.1K0
jscodeshift 那点秘密
copilot源码详细分析(三)ghostText核心逻辑
代码补全逻辑入口在calculateInlineCompletions 这个函数中:
孟健
2023/11/16
4870
copilot源码详细分析(三)ghostText核心逻辑
Python Ast介绍及应用
Abstract Syntax Trees即抽象语法树。Ast是python源码到字节码的一种中间产物,借助ast模块可以从语法树的角度分析源码结构。此外,我们不仅可以修改和执行语法树,还可以将Source生成的语法树unparse成python源码。因此ast给python源码检查、语法分析、修改代码以及代码调试等留下了足够的发挥空间。
py3study
2020/02/10
2.4K1
手写第一个 babel 插件
在前文《babel是怎样工作的》中介绍了 Bable 中的的AST,这次咱们给 bable 写一个插件,文中会覆盖大部份的用法,如果你对某些细节不是很明白,可以去看一下官方的 Babel 手册,中文版在这里:
疯狂的技术宅
2020/12/31
8410
手写第一个 babel 插件
一文助你搞懂 AST
抽象语法树(Abstract Syntax Tree)简称 AST,是源代码的抽象语法结构的树状表现形式。webpack、eslint 等很多工具库的核心都是通过抽象语法书这个概念来实现对代码的检查、分析等操作。今天我为大家分享一下 JavaScript 这类解释型语言的抽象语法树的概念
coder_koala
2020/06/22
2.7K1
【源码解析+代码实现】一篇文章搞定 babel-plugin-import 插件
平时在使用 antd、element 等组件库的时候,都会使用到一个 Babel 插件:babel-plugin-import,这篇文章通过例子和分析源码简单说一下这个插件做了一些什么事情,并且实现一个最小可用版本。
前端进击者
2021/07/27
1.7K0
【图文详解】200行JS代码,带你实现代码编译器(人人都能学会)
[011ead2e167b86d1d4def84147fbbdf6c6bb1c01.jpg] 最近看到掘金、前端公众号好多 ES2020 的文章,想说一句:放开我,我还学得动! 先问大家一句,日常项目开发中你能离开 ES6 吗? 一、前言 对于前端同学来说,编译器可能适合神奇的魔盒🎁,表面普通,但常常给我们惊喜。 编译器,顾名思义,用来编译,编译什么呢?当然是编译代码咯🌹。 [20191125-144728-7a47.gif] 其实我们也经常接触到编译器的使用场景: React 中 JSX 转换成
pingan8787
2020/03/29
3.3K0
【图文详解】200行JS代码,带你实现代码编译器(人人都能学会)
手摸手实现一个编译器(上)
PEG.js 是一个简单的 JavaScript 解析器生成器,可以生成具有出色错误报告的快速解析器。您可以使用它来处理复杂的数据或计算机语言,并轻松构建转换器、解释器、编译器和其他工具。
码农小余
2022/06/16
8190
手摸手实现一个编译器(上)
跨端框架模板解析引擎了解一下
Chameleon作为一个优秀的跨多端框架,想要实现"跨多端"这个核心目标,除了工程化配置之外,DSL层面对于各个端的转化的能力也是必不可少的,接下来将为大家介绍CML是如何做DSL层面的转化的。由于这部分只涉及到模板部分,也就是template标签中的内容转化,接下来所有提到的DSL都指的是模板内容。想了解更多有关CML跨端解决方案的内容,请访问 https://cml.didi.cn
腾讯云开发TCB
2020/07/30
6040
跨端框架模板解析引擎了解一下
TVM 学习指南(个人版)
最近粗略的看完了天奇大佬的MLC课程(顺便修了一些语法和拼写错误,也算是做了微弱的贡献hh),对TVM的近期发展有了一些新的认识。之前天奇大佬在《新一代深度学习编译技术变革和展望》一文中(链接:https://zhuanlan.zhihu.com/p/446935289)讲解了TVM Unify也即统一多层抽象的概念。这里的统一多层抽象具体包括AutoTensorization用来解决硬件指令声明和张量程序对接,TVM FFI(PackedFunc)机制使得我们可以灵活地引入任意的算子库和运行库函数并且在各个编译模块和自定义模块里面相互调用。TensorIR负责张量级别程序和硬件张量指令的整合。Relax (Relax Next) 引入relay的进一步迭代,直接引入first class symbolic shape的支持 (摘抄自《新一代深度学习编译技术变革和展望》一文)。然后这些抽象可以相互交互和联合优化来构造深度学习模型对应的最终部署形式。我个人感觉TVM Unify类似于MLIR的Dialect,但是这几个抽象的直接交互能力相比于MLIR的逐级lower我感觉是更直观方便的,毕竟是Python First(这个只是我最近看MLC课程的一个感觉)。对这部分内容感兴趣的读者请查看天奇大佬的TVM Unify介绍原文以及MLC课程。
BBuf
2022/09/28
3.9K0
TVM 学习指南(个人版)
webpack 基础知识整理
webpack是一个 模块打包工具,支持所有的打包语法,比如 ES Module、CommonJS、CMD、AMD。初期的webpack是用来模块打包js的,发展到现在,已经可以打包很多种文件类型,比如 css、img 。
神葳
2021/01/22
1.4K0
PyTorch 源码解读之即时编译篇
来源丨https://zhuanlan.zhihu.com/p/361101354
BBuf
2021/07/01
1.3K0
PyTorch 源码解读之即时编译篇
阿里面试官:如何给所有的async函数添加try/catch?
后来花了一些时间整理了下思路,那么如何实现给所有的async函数添加try/catch呢?
coder_koala
2022/11/28
1.1K0
阿里面试官:如何给所有的async函数添加try/catch?
webpack-bundle-analyzer 核心实现解析
webpack-bundle-analyzer 是一个插件:通过分析构建产物,最终生成 矩形树图 方便开发者根据项目构建后的依赖关系以及实际的文件尺寸,来进行相应的性能优化。
上山打老虎了
2022/06/15
3.1K0
webpack-bundle-analyzer 核心实现解析
带着问题读 TiDB 源码:Power BI Desktop 以 MySQL 驱动连接 TiDB 报错
常有人说,阅读源码是每个优秀开发工程师的必经之路,但是在面对像类似 TiDB 这样复杂的系统时,源码阅读是一个非常庞大的工程。而对一些 TiDB User 来说,从自己日常遇到的问题出发,反过来阅读源码就是一个不错的切入点,因此我们策划了《带着问题读源码》系列文章。
PingCAP
2021/12/02
8580
将JSON对象格式化输出(上篇)
(这后面的内容不用看了) 最近在使用Node.js开发项目,由于JavaScript内置了对JSON的支持,自然而然想到了使用JSON编写配置文件。
用户3579639
2018/10/19
2.1K0
将JSON对象格式化输出(上篇)
推荐阅读
相关推荐
花了大半个月,我终于逆向分析了Github Copilot
更多 >
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
本文部分代码块支持一键运行,欢迎体验
本文部分代码块支持一键运行,欢迎体验