百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

使用opencv实现深度学习的图片与视频的超分辨率

bigegpt 2024-08-02 10:53 8 浏览

什么是视频与图片的超分辨率,总结一下便是给一张分辨率比较低的图片,进行超分辨率的处理后,生成比较清晰的高分辨率的图片,上图图片完美解释了超分辨率的过程,由于不同的算法不同,处理的结果也不相同,本期我们介绍一下如何进行图片的超分辨率的处理。

· EDSR 模型

图像超分辨率EDSR:Enhanced Deep Residual Networks for Single Image Super-Resolution是 韩国首尔大学研究团队提出的用于图像超分辨率任务的新方法,分别用于增强深度超分辨率网络(enhanced deep super-resolution network,EDSR)和 实现一种新的多尺度深度超分辨率(multi-scale deep super-resolution system,MDSR),其在减小模型大小的同时实现了比当前其他方法更好的性能,分别赢得NTIRE2017超分辨率挑战赛的第一名和第二名。

EDSR的网络结构

如上图,和SRResnet相似,在残差块外,不经过relu层,并且去掉正则化层。

Single-scale model

太多的残差块会导致训练不稳定,文中使用了residual scaling的方法,即残差块在相加前,经过卷积处理的一路乘以一个小数scaling factor 0.1,保证训练更加稳定。

Multi-scale model

多尺度模型,一开始每个尺度都有两个独自的残差块,之后经过若干个残差块,最后再用独自的升采样模块来提高分辨率。

关于EDSR的模型,其官方已经训练好了预训练模型,我们可以直接使用OpenCV函数进行模型的加载与使用,接下来直接上代码

这里需要注意的是cv2.dnn_superres函数存在于OpenCV4.4以上以及,需要更新一下自己的OpenCV版本。其函数支持4种模型:

  • EDSR
  • ESPCN
  • FSRCNN
  • LapSRN

除了以上4种模型外,OpenCV还提供了bilinear, bicubic 2种方法

import time
import cv2
import os
modelName = 'EDSR_x2.pb'.split(os.path.sep)[-1].split("_")[0].lower()
modelScale = 'model/EDSR_x2.pb'.split("_x")[-1]
modelScale = int(modelScale[:modelScale.find(".")])
print("[INFO] loading super resolution model: {}".format(
'model/EDSR_x2.pb'))
print("[INFO] model name: {}".format(modelName))
print("[INFO] model scale: {}".format(modelScale))
sr = cv2.dnn_superres.DnnSuperResImpl_create()
sr.readModel('model/EDSR_x2.pb')
sr.setModel(modelName, modelScale)

这里首先我们需要设置一下模型的基础信息,获取模型的名称以及需要放大的倍数,然后使用cv2.dnn_superres.DnnSuperResImpl_create函数定义一个超分辨率的方法,使用readModel函数对模型进行读取,并使用setModel函数对模型进行设置

image = cv2.imread('comic.png')
print("[INFO] w: {}, h: {}".format(image.shape[1], image.shape[0]))
start = time.time()
upscaled = sr.upsample(image)
end = time.time()
print("[INFO] super resolution took {:.6f} seconds".format(
end - start))
print("[INFO] w: {}, h: {}".format(upscaled.shape[1],
upscaled.shape[0]))

我们读取一张需要进行处理的照片,并设置一下初始时间与结束时间,以便对比多个模型的处理时间,这里直接使用sr.upsample(image)函数,进行图片的超分辨率操作

start = time.time()
bicubic = cv2.resize(image, (upscaled.shape[1], upscaled.shape[0]),
interpolation=cv2.INTER_CUBIC)
#bicubic = cv2.resize(img, None, fx=scale, fy=scale, interpolation=cv2.INTER_LINEAR)
end = time.time()
print("[INFO] bicubic interpolation took {:.6f} seconds".format(
end - start))
cv2.imshow("Original", image)
cv2.imwrite("Bicubic1.png", bicubic)
cv2.imwrite("Super Resolution1.png", upscaled)
cv2.waitKey(0)

这里我们同步添加了OpenCV的bicubic以及linear算法,从结果可以看出,OpenCV提供的算法速度很快,但是效果一般

LapSRN是2017年在CVPR上发表的超分辨率算法,作者结合传统图像算法拉普拉斯金字塔与深度学习,实现多级超分辨模型,由LR生成不同超分辨率SR。

优点:1)速度快:和FSRCNN差不多,实现了实时重建

2)精度高:LapSRN直接从LR图像提取特征图,用深的卷积层优化上采样过滤器并预测子带残差

二、基本原理:

模型有两个分支,上面是特征提取分支,下面是图像重构分支。

(红色箭头:卷积进行特征提取,蓝色箭头:反卷积提升分辨率,绿色箭头:两张图片逐像素相加)

上层分支即特征提取中,每一级有d个卷积层,再跟随一个2倍的反卷积提升分辨率即图像长宽各扩大一倍,每个反卷积层的输出与两个不同层链接:(1) 重建残差图像的卷积层;(2)s+1层提取特征的卷积层。残差图像(图中黑白图像)和采用相同方式(上采样)提升分辨率的LR图像相加,得到更精细的SR。这样采用逐层递进的方式可以重建出更为清晰的超分辨图像,也因此可以获得不同scale的重构图像。

三、模型框架:

我们修改一下EDSR模型的代码,直接读取LapSRN模型

import time
import cv2
import os
modelName = 'LapSRN_x2.pb'.split(os.path.sep)[-1].split("_")[0].lower()
modelScale = 'model/LapSRN_x2.pb'.split("_x")[-1]
modelScale = int(modelScale[:modelScale.find(".")])
print("[INFO] loading super resolution model: {}".format(
'model/LapSRN_x2.pb'))
print("[INFO] model name: {}".format(modelName))
print("[INFO] model scale: {}".format(modelScale))
sr = cv2.dnn_superres.DnnSuperResImpl_create()
sr.readModel('model/LapSRN_x2.pb')
sr.setModel(modelName, modelScale)
image = cv2.imread('comic.png')
print("[INFO] w: {}, h: {}".format(image.shape[1], image.shape[0]))
start = time.time()
upscaled = sr.upsample(image)
end = time.time()
print("[INFO] super resolution took {:.6f} seconds".format(
end - start))
print("[INFO] w: {}, h: {}".format(upscaled.shape[1],
upscaled.shape[0]))
start = time.time()
bicubic = cv2.resize(image, (upscaled.shape[1], upscaled.shape[0]),
interpolation=cv2.INTER_CUBIC)
#bicubic = cv2.resize(img, None, fx=scale, fy=scale, interpolation=cv2.INTER_LINEAR)
end = time.time()
print("[INFO] bicubic interpolation took {:.6f} seconds".format(
end - start))
cv2.imshow("Original", image)
cv2.imwrite("Bicubic11.png", bicubic)
cv2.imwrite("Super Resolution11.png", upscaled)
cv2.waitKey(0)

这里我们只需要修改源代码中的模型文件地址以及模型文件即可,对比来讲LapSRN虽然精度有点欠缺,但是速度很快

[INFO] loading super resolution model: model/LapSRN_x2.pb
[INFO] model name: lapsrn
[INFO] model scale: 2
[INFO] w: 250, h: 361
[INFO] super resolution took 0.700706 seconds
[INFO] w: 500, h: 722
[INFO] bicubic interpolation took 0.001995 seconds

处理一张下图的照片仅用0.7s,但是EDSR模型至少需要5S 以上的时间,EDSR虽然精度有改善,但是时间太长,若对时间要求比较紧张的场合,可以考虑使用LapSRN模型来替代。

关于其他模型参考如下图片,其跟EDSR模型类似,但是速度很快,当然要牺牲一些精度,还有其他2种模型,这里不再试运行,感兴趣的小伙伴可以自行尝试,对比效果

OpenCV中的dnn_superres模块提供的四种图像超分放大深度学习模型,在实践中用得最多的就是EDSR模型。其他三类模型和OpenCV自带的resize函数视觉上差别并不大,当然也可以直接使用OpenCV提供的bicubic以及bilinear。但是EDSR模型推理速度太慢,2倍放大和4倍放大可以使用ESPCN代替,4倍和8倍放大可以使用LapSRN。其OpenCV官方发布了几种模型的对比如下图

2种模型的速度很快,但精度较差,这里我们主要使用的还是EDSR与LapSRN模型,当然随着深度学习的不断加入,越来越多的算法能够实现超分辨率的效果,当我们遇到比较模糊的照片时,在提高分辨率的同时我们更希望提高图片的清晰度,可以看到以上算法只是提高了图片的分辨率,并没有提高图片的清晰度,下期我们将分享其他的超分辨率的算法。

相关推荐

当Frida来“敲”门(frida是什么)

0x1渗透测试瓶颈目前,碰到越来越多的大客户都会将核心资产业务集中在统一的APP上,或者对自己比较重要的APP,如自己的主业务,办公APP进行加壳,流量加密,投入了很多精力在移动端的防护上。而现在挖...

服务端性能测试实战3-性能测试脚本开发

前言在前面的两篇文章中,我们分别介绍了性能测试的理论知识以及性能测试计划制定,本篇文章将重点介绍性能测试脚本开发。脚本开发将分为两个阶段:阶段一:了解各个接口的入参、出参,使用Python代码模拟前端...

Springboot整合Apache Ftpserver拓展功能及业务讲解(三)

今日分享每天分享技术实战干货,技术在于积累和收藏,希望可以帮助到您,同时也希望获得您的支持和关注。架构开源地址:https://gitee.com/msxyspringboot整合Ftpserver参...

Linux和Windows下:Python Crypto模块安装方式区别

一、Linux环境下:fromCrypto.SignatureimportPKCS1_v1_5如果导包报错:ImportError:Nomodulenamed'Crypt...

Python 3 加密简介(python des加密解密)

Python3的标准库中是没多少用来解决加密的,不过却有用于处理哈希的库。在这里我们会对其进行一个简单的介绍,但重点会放在两个第三方的软件包:PyCrypto和cryptography上,我...

怎样从零开始编译一个魔兽世界开源服务端Windows

第二章:编译和安装我是艾西,上期我们讲述到编译一个魔兽世界开源服务端环境准备,那么今天跟大家聊聊怎么编译和安装我们直接进入正题(上一章没有看到的小伙伴可以点我主页查看)编译服务端:在D盘新建一个文件夹...

附1-Conda部署安装及基本使用(conda安装教程)

Windows环境安装安装介质下载下载地址:https://www.anaconda.com/products/individual安装Anaconda安装时,选择自定义安装,选择自定义安装路径:配置...

如何配置全世界最小的 MySQL 服务器

配置全世界最小的MySQL服务器——如何在一块IntelEdison为控制板上安装一个MySQL服务器。介绍在我最近的一篇博文中,物联网,消息以及MySQL,我展示了如果Partic...

如何使用Github Action来自动化编译PolarDB-PG数据库

随着PolarDB在国产数据库领域荣膺桂冠并持续获得广泛认可,越来越多的学生和技术爱好者开始关注并涉足这款由阿里巴巴集团倾力打造且性能卓越的关系型云原生数据库。有很多同学想要上手尝试,却卡在了编译数据...

面向NDK开发者的Android 7.0变更(ndk android.mk)

订阅Google官方微信公众号:谷歌开发者。与谷歌一起创造未来!受Android平台其他改进的影响,为了方便加载本机代码,AndroidM和N中的动态链接器对编写整洁且跨平台兼容的本机...

信创改造--人大金仓(Kingbase)数据库安装、备份恢复的问题纪要

问题一:在安装KingbaseES时,安装用户对于安装路径需有“读”、“写”、“执行”的权限。在Linux系统中,需要以非root用户执行安装程序,且该用户要有标准的home目录,您可...

OpenSSH 安全漏洞,修补操作一手掌握

1.漏洞概述近日,国家信息安全漏洞库(CNNVD)收到关于OpenSSH安全漏洞(CNNVD-202407-017、CVE-2024-6387)情况的报送。攻击者可以利用该漏洞在无需认证的情况下,通...

Linux:lsof命令详解(linux lsof命令详解)

介绍欢迎来到这篇博客。在这篇博客中,我们将学习Unix/Linux系统上的lsof命令行工具。命令行工具是您使用CLI(命令行界面)而不是GUI(图形用户界面)运行的程序或工具。lsoflsof代表&...

幻隐说固态第一期:固态硬盘接口类别

前排声明所有信息来源于网络收集,如有错误请评论区指出更正。废话不多说,目前固态硬盘接口按速度由慢到快分有这几类:SATA、mSATA、SATAExpress、PCI-E、m.2、u.2。下面我们来...

新品轰炸 影驰SSD多款产品登Computex

分享泡泡网SSD固态硬盘频道6月6日台北电脑展作为全球第二、亚洲最大的3C/IT产业链专业展,吸引了众多IT厂商和全球各地媒体的热烈关注,全球存储新势力—影驰,也积极参与其中,为广大玩家朋友带来了...