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

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

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

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

· 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模型,当然随着深度学习的不断加入,越来越多的算法能够实现超分辨率的效果,当我们遇到比较模糊的照片时,在提高分辨率的同时我们更希望提高图片的清晰度,可以看到以上算法只是提高了图片的分辨率,并没有提高图片的清晰度,下期我们将分享其他的超分辨率的算法。

相关推荐

得物可观测平台架构升级:基于GreptimeDB的全新监控体系实践

一、摘要在前端可观测分析场景中,需要实时观测并处理多地、多环境的运行情况,以保障Web应用和移动端的可用性与性能。传统方案往往依赖代理Agent→消息队列→流计算引擎→OLAP存储...

warm-flow新春版:网关直连和流程图重构

本期主要解决了网关直连和流程图重构,可以自此之后可支持各种复杂的网关混合、多网关直连使用。-新增Ruoyi-Vue-Plus优秀开源集成案例更新日志[feat]导入、导出和保存等新增json格式支持...

扣子空间体验报告

在数字化时代,智能工具的应用正不断拓展到我们工作和生活的各个角落。从任务规划到项目执行,再到任务管理,作者深入探讨了这款工具在不同场景下的表现和潜力。通过具体的应用实例,文章展示了扣子空间如何帮助用户...

spider-flow:开源的可视化方式定义爬虫方案

spider-flow简介spider-flow是一个爬虫平台,以可视化推拽方式定义爬取流程,无需代码即可实现一个爬虫服务。spider-flow特性支持css选择器、正则提取支持JSON/XML格式...

solon-flow 你好世界!

solon-flow是一个基础级的流处理引擎(可用于业务规则、决策处理、计算编排、流程审批等......)。提供有“开放式”驱动定制支持,像jdbc有mysql或pgsql等驱动,可...

新一代开源爬虫平台:SpiderFlow

SpiderFlow:新一代爬虫平台,以图形化方式定义爬虫流程,不写代码即可完成爬虫。-精选真开源,释放新价值。概览Spider-Flow是一个开源的、面向所有用户的Web端爬虫构建平台,它使用Ja...

通过 SQL 训练机器学习模型的引擎

关注薪资待遇的同学应该知道,机器学习相关的岗位工资普遍偏高啊。同时随着各种通用机器学习框架的出现,机器学习的门槛也在逐渐降低,训练一个简单的机器学习模型变得不那么难。但是不得不承认对于一些数据相关的工...

鼠须管输入法rime for Mac

鼠须管输入法forMac是一款十分新颖的跨平台输入法软件,全名是中州韵输入法引擎,鼠须管输入法mac版不仅仅是一个输入法,而是一个输入法算法框架。Rime的基础架构十分精良,一套算法支持了拼音、...

Go语言 1.20 版本正式发布:新版详细介绍

Go1.20简介最新的Go版本1.20在Go1.19发布六个月后发布。它的大部分更改都在工具链、运行时和库的实现中。一如既往,该版本保持了Go1的兼容性承诺。我们期望几乎所...

iOS 10平台SpriteKit新特性之Tile Maps(上)

简介苹果公司在WWDC2016大会上向人们展示了一大批新的好东西。其中之一就是SpriteKitTileEditor。这款工具易于上手,而且看起来速度特别快。在本教程中,你将了解关于TileE...

程序员简历例句—范例Java、Python、C++模板

个人简介通用简介:有良好的代码风格,通过添加注释提高代码可读性,注重代码质量,研读过XXX,XXX等多个开源项目源码从而学习增强代码的健壮性与扩展性。具备良好的代码编程习惯及文档编写能力,参与多个高...

Telerik UI for iOS Q3 2015正式发布

近日,TelerikUIforiOS正式发布了Q32015。新版本新增对XCode7、Swift2.0和iOS9的支持,同时还新增了对数轴、不连续的日期时间轴等;改进TKDataPoin...

ios使用ijkplayer+nginx进行视频直播

上两节,我们讲到使用nginx和ngixn的rtmp模块搭建直播的服务器,接着我们讲解了在Android使用ijkplayer来作为我们的视频直播播放器,整个过程中,需要注意的就是ijlplayer编...

IOS技术分享|iOS快速生成开发文档(一)

前言对于开发人员而言,文档的作用不言而喻。文档不仅可以提高软件开发效率,还能便于以后的软件开发、使用和维护。本文主要讲述Objective-C快速生成开发文档工具appledoc。简介apple...

macOS下配置VS Code C++开发环境

本文介绍在苹果macOS操作系统下,配置VisualStudioCode的C/C++开发环境的过程,本环境使用Clang/LLVM编译器和调试器。一、前置条件本文默认前置条件是,您的开发设备已...