提升视频的分辨率

在本教程中，你将学习如何利用「dnn_superres」接口通过预先训练的神经网络提升视频的分辨率。

构建

在构建 OpenCV 时，运行以下命令构建所有的 contrib 模块

cmake -D OPENCV_EXTRA_MODULES_PATH=<opencv_contrib>/modules/

或仅构建 dnn_superres 模块

cmake -D OPENCV_EXTRA_MODULES_PATH=<opencv_contrib>/modules/dnn_superres

或确保你在 CMake 图形界面版本中勾选了 dnn_superres 模块：cmake-gui。

示例的源代码

// 此文件是 OpenCV 项目的一部分。
// 它遵循位于该发行包的顶级目录和 https://opencv.ac.cn/license.html 的 LICENSE 文件中的许可条款。
// of this distribution and at https://opencv.ac.cn/license.html.
 
#include <iostream>
 
#include <opencv2/dnn_superres.hpp>
 
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
 
using namespace std;
using namespace cv;
using namespace dnn_superres;
 
int main(int argc, char *argv[])
{
 // 检查有效的命令行参数，如果参数数量不足，则打印用法。
 // if insufficient arguments were given.
 if (argc < 4) {
 cout << "usage: Arg 1: input video path" << endl;
 cout << "\t Arg 2: output video path" << endl;
 cout << "\t Arg 3: 算法 | edsr、espcn、fsrcnn 或 lapsrn" << endl;
 cout << "\t Arg 4: 比例 | 2、3、4 或 8 \n";
 cout << "\t Arg 5: 模型文件路径 \n";
 return -1;
    }
 
 string input_path = string(argv[1]);
 string output_path = string(argv[2]);
 string algorithm = string(argv[3]);
 int scale = atoi(argv[4]);
 string path = string(argv[5]);
 
 VideoCapture input_video(input_path);
 int ex = static_cast<int>(input_video.get(CAP_PROP_FOURCC));
 Size S = Size((int) input_video.get(CAP_PROP_FRAME_WIDTH) * scale,
 (int) input_video.get(CAP_PROP_FRAME_HEIGHT) * scale);
 
 VideoWriter output_video;
 output_video.open(output_path, ex, input_video.get(CAP_PROP_FPS), S, true);
 
 if (!input_video.isOpened())
    {
 std::cerr << "无法打开视频。" << std::endl;
 return -1;
    }
 
 DnnSuperResImpl sr;
 sr.readModel(path);
 sr.setModel(algorithm, scale);
 
 for(;;)
    {
 Mat frame, output_frame;
 input_video >> frame;
 
 如果 ( frame。empty() )
 中断;
 
 sr.upsample(frame, output_frame);
 output_video << output_frame;
 
 namedWindow("插值视频", WINDOW_AUTOSIZE);
 imshow("插值视频", output_frame);
 
 namedWindow("原始视频", WINDOW_AUTOSIZE);
 imshow("原始视频", frame);
 
 字符 c=(字符)waitKey(25);
 如果(c==27)
 中断;
    }
 
 input_video.release();
 output_video.release();
 
 返回 0;
}

说明

1. 设置报头和命名空间

   #include <opencv2/dnn_superres.hpp>
   using namespace std;
   using namespace cv;
   using namespace dnn_superres;

2. 创建 Dnn Superres 对象

   DnnSuperResImpl sr;

   实例化一个 dnn 超分辨率对象。
3. 读取模型

   path = "models/ESPCN_x2.pb"
   sr.readModel(path);
   sr.setModel("espcn", 2);

   从给定路径中读取模型，并设置算法和缩放因子。
4. 放大视频

   对于(;;)
   {
       Mat frame, output_frame;
   input_video >> frame;
       if ( frame.empty() )
           中断;
   sr.upsample(frame, output_frame);
       ...
   }

   逐帧处理并放大视频。