图像处理入门

目标

在本教程中，你将学习如何：

• 加载图像（使用 cv::imread）

• 在一个 OpenCV 窗口中显示一张图像（使用 cv::imshow）

• 将图像写入文件（使用 cv::imwrite）

源代码

#include <iostream>
#include <string>
#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/highgui.hpp>

using namespace cv;

int main()
{
    std::string image_path = samples::findFile("starry_night.jpg");
    Mat img = imread(image_path, IMREAD_COLOR);

    if (img.empty()) {
        std::cout << "Could not read the image: " << image_path << std::endl;
        return 1;
    }

    imshow("Display window", img);
    int k = waitKey(0); // 等待用户按下按键

    if (k == 's')
        imwrite("starry_night.png", img);
   
    return 0;
}

代码解释

在 OpenCV 3 中，我们有多个模块。每个模块都负责图像处理的不同的领域或方法。你可以在这个教程本身的用户指南的结构中观察到这一点。在使用它们之前，首先需要包括声明了每个模块内容的头文件。

你几乎总是会用到：

• core 模块，它定义了库的基本结构
• imgcodecs 模块，它提供了读写函数
• highgui 模块，因为它包含在窗口中显示图像的函数

我们还应包括 iostream，以方便控制台的输入和输出。

以下通过声明 using namespace cv;，无需显式书写命名空间即可访问库函数。

#include <opencv2/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/highgui.hpp>

#include <iostream>

using namespace cv;

现在，让我们来分析主要代码。 第一步，我们使用 OpenCV samples 中的 findFile 来获取图像 “ starry_night.jpg” 的路径。 接着，调用 cv::imread 函数来读取图像，它的第一个参数用来指定的图像的文件路径， 第二个参数是可选的，用来指定我们想要图像的格式。它可以是：

• IMREAD_COLOR 将图像加载为 BGR 8-bit 的格式。这是此处使用的默认值。
• IMREAD_UNCHANGED 按原样加载图像（包括 alpha 通道（如果存在的话））。
• IMREAD_GRAYSCALE 将图像加载为灰度图。

读入图像后，数据将存储在 cv::Mat 对象中。

std::string image_path = samples::findFile("starry_night.jpg");
Mat img = imread(image_path, IMREAD_COLOR);

注意：OpenCV 支持图像格式有 Windows 位图（bmp），可移植图像格式（pbm，pgm，ppm）和 Sun 栅格图（sr，ras）。在插件的帮助下（如果你自己构建库，则需要指定使用它们，但是在我们默认提供的包中）你还可以加载 JPEG（jpeg，jpg，jpe），JPEG 2000（jp2 - 在 CMake 代号为 Jasper），TIFF 文件（tiff，tif）和便携式网络图形（png）。此外，OpenEXR 格式也可能支持。

然后，检查图像是否正确被加载。 

if (img.empty()) {
    std::cout << "Could not read the image: " << image_path << std::endl;
    return 1;
}

然后，通过调用 cv::imshow 函数显示图像。第一个参数是窗口的标题，第二个参数是将要显示的 cv::Mat 对象。

因为我们希望窗口一直显示到用户按下一个键（否则程序将很快结束），所以我们使用 cv::waitKey 函数，它唯一的参数是等待用户输入的时间（以毫秒为单位）。零意味着永远等待。返回值是被按下的键。

imshow("Display window", img);
int k = waitKey(0); // 等待用户按下按键

最后，如果按下的键是“s”键，则将图像写入文件。为此，将调用 cv::imwrite 函数，该函数使用文件路径和 cv::Mat 对象作为参数。