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中的try和finally和with方法
May 05 Python
Python的Django框架中URLconf相关的一些技巧整理
Jul 18 Python
Python实现判断字符串中包含某个字符的判断函数示例
Jan 08 Python
Pycharm设置界面全黑的方法
May 23 Python
利用Pyhton中的requests包进行网页访问测试的方法
Dec 26 Python
python2和python3在处理字符串上的区别详解
May 29 Python
python读写csv文件方法详细总结
Jul 05 Python
Python网络编程之使用TCP方式传输文件操作示例
Nov 01 Python
python实现UDP协议下的文件传输
Mar 20 Python
python 比较字典value的最大值的几种方法
Apr 17 Python
查看keras的默认backend实现方式
Jun 19 Python
python 三种方法实现对Excel表格的读写
Nov 19 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
php递归列出所有文件和目录的代码
2008/09/10 PHP
php最简单的删除目录与文件实现方法
2014/11/28 PHP
php代码检查代理ip的有效性
2016/08/19 PHP
php组合排序简单实现方法
2016/10/15 PHP
img的onload的另类用法
2008/01/10 Javascript
Cookie 小记
2010/04/01 Javascript
在Windows上安装Node.js模块的方法
2011/09/25 Javascript
javascript设置连续两次点击按钮时间间隔的方法
2014/10/28 Javascript
AngularJs  Creating Services详解及示例代码
2016/09/02 Javascript
Javascript实现的StopWatch功能示例
2017/06/13 Javascript
layui.js实现的表单验证功能示例
2017/11/15 Javascript
利用Console来Debug的10个高级技巧汇总
2018/03/26 Javascript
使用javascript函数编写简单银行取钱存钱流程
2018/05/26 Javascript
Puppet的一些技巧
2018/09/17 Javascript
vue-cli 项目打包完成后运行文件路径报错问题
2019/07/19 Javascript
javascript设计模式之迭代器模式
2020/01/30 Javascript
10 分钟快速入门 Python3的教程
2019/01/29 Python
Python Opencv任意形状目标检测并绘制框图
2019/07/23 Python
Django admin model 汉化显示文字的实现方法
2019/08/12 Python
简单了解python shutil模块原理及使用方法
2020/04/28 Python
python pandas dataframe 去重函数的具体使用
2020/07/20 Python
CSS3 :nth-child()伪类选择器实现奇偶行显示不同样式
2013/11/05 HTML / CSS
eDreams葡萄牙:全球最大的在线旅行社之一
2019/04/15 全球购物
美国时尚大码女装购物网站:Avenue
2019/05/24 全球购物
乌克兰最大的家用电器和电子产品连锁店:Eldorado
2019/10/02 全球购物
暑期实践思想汇报
2014/01/06 职场文书
高中考试作弊检讨书
2014/01/14 职场文书
无私奉献演讲稿
2014/09/04 职场文书
食品委托检验协议书范本
2014/09/12 职场文书
忠诚与背叛观后感
2015/06/04 职场文书
地心历险记观后感
2015/06/15 职场文书
关于党风廉政建设宣传教育月的活动总结!
2019/08/08 职场文书
python基于OpenCV模板匹配识别图片中的数字
2021/03/31 Python
PyQt5结合QtDesigner实现文本框读写操作
2021/06/11 Python
Html5获取用户当前位置的几种方式
2022/01/18 HTML / CSS
在MySQL中你成功的避开了所有索引
2022/04/20 MySQL