Python 视频画质增强


Posted in Python onApril 28, 2022

前言

前面通过文章 Python捕获、播放和保存摄像头视频并提高视频清晰度和对比度给大家介绍了如何读取、播放和保存视频,后面又通过文章 Python美图技术也就几行代码!给大家介绍了如何对图像的亮度、对比度、色度或者锐度进行调整,从而达到基本的图像处理操作。

其实,那两篇文章都是铺垫,都是为了给大家介绍如何对视频进行画质增强。本文将结合前面两篇文章的内容,来讲讲如何对视频画质进行增强。

想要直接看效果的,可以拉到文末。

原理

不知道大家小时候有没有玩过这个?

Python 视频画质增强

最早的动画就是这么形成的,记得小时候还有这种小书卖。

其实视频的原理也是这样,一个视频是由很多张图片组成的,一个图片是一帧。所以我们要对视频进行画质增强,可以拆分成对每一帧的图片进行操作,这个操作我们在前面介绍过。

因此,对视频进行画质增强的方法可以分为三步:拆分->处理->合成。

实现步骤

拆分

我们在第一篇文章讲到过如何捕获摄像头的视频流,以及如何读取视频并播放。不管通过哪种方法,我们都是通过帧操作的。所以这里所谓的拆分就是获取到我们捕获到的视频流或者读取的视频流的每一帧。

success, img1 = cap.read()
    # 如果正确读取帧,success为True
    if not success:
        break
    cv2.imshow('img1', img1)

就是这么简单,我们就可以获取到视频每一帧了。

处理

获取到视频的一帧之后,我们就要把这一帧转换成我们可以处理的格式的图片。在前面我们介绍如何对图片进行画质增强的时候,使用的是 ImageEnhance 这个函数的相关方法,这个函数是 PIL 图像处理库里面的,所以我们必须把我们每一帧的图片读取成 PIL 可以处理的格式:

image = Image.fromarray(np.uint8(img1))  # 转换成PIL可以处理的格式

读取到图像之后,我们就可以对图像进行画质增强处理了,这里还是用我们上篇文章中讲到的代码:

# 图像处理
def img_enhance(image, brightness=1, color=1,contrast=1,sharpness=1):
    # 亮度增强
    enh_bri = ImageEnhance.Brightness(image)
    if brightness:
        image = enh_bri.enhance(brightness)

    # 色度增强
    enh_col = ImageEnhance.Color(image)
    if color:
        image = enh_col.enhance(color)

    # 对比度增强
    enh_con = ImageEnhance.Contrast(image)
    if contrast:
        image = enh_con.enhance(contrast)

    # 锐度增强
    enh_sha = ImageEnhance.Sharpness(image)
    if sharpness:
        image = enh_sha.enhance(sharpness)

    return image

合成

图像处理完,我们需要每一帧图像进行合成,从而得到我们最终的视频:

cap = cv2.VideoCapture('你的视频目录/xxx.mp4')
success, _ = cap.read()
# 分辨率-宽度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
# 分辨率-高度
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
# 总帧数
frame_counter = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
video_writer = cv2.VideoWriter('输出.mp4', cv2.VideoWriter_fourcc('M', 'P', '4', 'V'), 15, (width, height), True)

while success:
    success, img1 = cap.read()
    try:
        image = Image.fromarray(np.uint8(img1))  # 转换成PIL可以处理的格式
        img_enhanced = img_enhance(image, 2, 2, 2, 3)
        video_writer.write(np.asarray(img_enhanced))
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    except:
        break


cap.release()
video_writer.release()
cv2.destroyAllWindows()

我这里读取的是 mp4 格式的视频,所以在合成写视频文件的时候,我们需要用

cv2.VideoWriter_fourcc('M', 'P', '4', 'V') 这个格式。

我这里没有对图片的分辨率进行修改,只是分别获取原始视频的分辨率,然后写入视频文件的时候,将原始分辨率传入作为参数。

如果你需要修改视频的分辨率的话,可以使用下面的方式:

cv2.resize(src, dsize[, dst[, fx[, fy[, interpolation]]]])

使用 resize 方法就可以了:

resized = cv2.resize(img, (width, height), interpolation = cv2.INTER_AREA)

效果

我们先来看看处理前的视频:

Python 视频画质增强

处理之后(我这里处理比较随意,参数都是随意写的)的视频是这样子的:

Python 视频画质增强

总结

到此为止,我们的视频画质增强的功能算是基本实现了,代码也不复杂,加起来也就这么点。但是,如果要处理成自己满意的效果,还是需要下一番功夫去调参数,去优化。甚至针对每一帧可能传入的参数都不一样,这就需要各位自己去慢慢研究了。

Python 相关文章推荐
python网络编程学习笔记(三):socket网络服务器
Jun 09 Python
Python+selenium实现截图图片并保存截取的图片
Jan 05 Python
Python面向对象class类属性及子类用法分析
Feb 02 Python
Python实现分段线性插值
Dec 17 Python
django ManyToManyField多对多关系的实例详解
Aug 09 Python
Python编译成.so文件进行加密后调用的实现
Dec 23 Python
Python基于Twilio及腾讯云实现国际国内短信接口
Jun 18 Python
在Tensorflow中实现leakyRelu操作详解(高效)
Jun 30 Python
python爬虫爬取网页数据并解析数据
Sep 18 Python
如何用python写个模板引擎
Jan 14 Python
Pandas的数据过滤实现
Jan 15 Python
python设置 matplotlib 正确显示中文的四种方式
May 10 Python
python 单机五子棋对战游戏
python井字棋游戏实现人机对战
Apr 28 #Python
Python开发五子棋小游戏
Python简易开发之制作计算器
Apr 28 #Python
Python实现对齐打印 format函数的用法
Apr 28 #Python
python实现简单的三子棋游戏
Apr 28 #Python
Python内置类型集合set和frozenset的使用详解
You might like
神族 Protoss 历史背景
2020/03/14 星际争霸
Search Engine Friendly的URL设计
2006/10/09 PHP
利用php递归实现无限分类 格式化数组的详解
2013/06/08 PHP
win7下memCache的安装过程(具体操作步骤)
2013/06/28 PHP
基于laravel belongsTo使用详解
2019/10/18 PHP
javascript判断用户浏览器插件安装情况的代码
2011/01/01 Javascript
使用jquery为table动态添加行的实现代码
2011/03/30 Javascript
input输入框的自动匹配(原生代码)
2013/03/19 Javascript
node.js实现BigPipe详解
2014/12/05 Javascript
sails框架的学习指南
2014/12/22 Javascript
jQuery中$.each使用详解
2015/01/29 Javascript
jquery插件Jplayer使用方法简析
2016/04/22 Javascript
jQuery 操作input中radio的技巧
2016/07/18 Javascript
基于js中的原型、继承的一些想法
2016/08/10 Javascript
js获取地址栏中传递的参数(两种方法)
2017/02/08 Javascript
BootStrap导航栏问题记录
2017/07/31 Javascript
iview table高度动态设置方法
2018/03/14 Javascript
vue中$nextTick的用法讲解
2019/01/17 Javascript
微信小程序sessionid不一致问题解决
2019/08/30 Javascript
[00:02]DOTA2新版本使用PA至宝后暴击展示
2014/11/19 DOTA
python开发之字符串string操作方法实例详解
2015/11/12 Python
Python Paramiko模块的使用实际案例
2018/02/01 Python
对python中使用requests模块参数编码的不同处理方法
2018/05/18 Python
python实现自动发送邮件
2018/06/20 Python
详解Python字典的操作
2019/03/04 Python
详解Python3 基本数据类型
2019/04/19 Python
Python中单线程、多线程和多进程的效率对比实验实例
2019/05/14 Python
使用python自动追踪你的快递(物流推送邮箱)
2020/03/17 Python
Python实现CAN报文转换工具教程
2020/05/05 Python
资生堂美国官网:Shiseido美国
2016/09/02 全球购物
现代绅士日常奢侈品:Todd Snyder
2019/12/13 全球购物
面向对象编程OOP的优点
2013/01/22 面试题
环境整治工作方案
2014/05/18 职场文书
基层党组织整改方案
2014/10/25 职场文书
论文答辩开场白大全
2015/05/27 职场文书
ROS系统将python包编译为可执行文件的简单步骤
2021/07/25 Python